output = input.matmul(weight.t()) RuntimeError: expected scalar type Long but found Float 错误解决

在使用pytorch的nn.Linear时出现错误

 RuntimeError: expected scalar type Long but found Float

这里报错的原因是我的输入是Longtensor,因此这里将输入改为Floattensor数据类型就可以了。

你可能感兴趣的:(pytorch)