在使用pytorch的nn.Linear时出现错误
RuntimeError: expected scalar type Long but found Float
这里报错的原因是我的输入是Longtensor,因此这里将输入改为Floattensor数据类型就可以了。
在使用pytorch的nn.Linear时出现错误
RuntimeError: expected scalar type Long but found Float
这里报错的原因是我的输入是Longtensor,因此这里将输入改为Floattensor数据类型就可以了。