【PyTorch】Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

和网上找到的答案都不一样。

问题是因为我在__init__外面的函数中定义了一个卷积操作

然后就一直报错。

具体原因之后再查吧。