Pytorch 卷积函数nn.Conv2d()运行报错:RuntimeError: expected scalar type Long but found Float
这里写目录标题实验目的遇到的问题解决方法实验目的简单进行一个4*4 tensor经过卷积生成[4 , 2, 2]的tensorimport torchfrom torch import nnclass Net(): def __init__(self): self.conv1 = nn.Conv2d(1, 4, kernel_size = 3, stride=1) def forword(self, x): print(x.type())
原创
2022-02-28 09:55:41 ·
1174 阅读 ·
0 评论