如何使用 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()