如何用Pytorch将FloatTensor转换为ByteTensor？

Question

我是Pytorch和神经网络编程的新手，但是遇到了一个问题，我无法自行解决。我的数据是1和0的numpy数组。但是当我尝试训练我的网络时，出现此错误：

  

RuntimeError：类型为torch.ByteTensor的预期对象，但发现参数＃2'mat2'的类型为torch.FloatTensor

错误所在的行在我的网的正向方法中

x = self.fc1(x)

我已经尝试过这些方法来转换张量，但是仍然出现错误：

  

x = x.type（'torch.ByteTensor'）

和

  

x.byte（）

Answer 1

x.byte（）返回所需的内容，但这不是“就地”方法。尝试做：

x = x.byte()