理解卷积的问题(conv(data)[i] == conv(data [i] .unsqueeze(0))??)
import torch
import torch.nn as nn
data = torch.ones(3,3,6,6)
conv = nn.Conv2d(3, 16, kernel_size = 3, padding = 1)
print(data[0].unsqueeze(0).shape)
for i in range(3):
print((conv(data)[i] == conv(data[i].unsqueeze(0))).all())
结果:
torch.Size([1, 3, 6, 6])
tensor(False)
tensor(False)
tensor(False)
我认为它将打印为True,但最终会打印False。知道为什么吗?
import torch
import torch.nn as nn
data = torch.ones(3,3,6,6)
conv = nn.Conv2d(3, 16, kernel_size = 3, padding = 1)
print(data[0].unsqueeze(0).shape)
for i in range(3):
print((conv(data)[i] == conv(data[i].unsqueeze(0))).all())
Results:
torch.Size([1, 3, 6, 6])
tensor(False)
tensor(False)
tensor(False)
I thought it would print True but ended up printing False instead. Any idea why?
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论