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

错误报在了forward里的Conv2d处。原因是函数写在forward里可能默认cpu，如果写在init构造函数里，就不需要再指定cuda。

修改为箭头指示就不再报错了。