报错信息
RuntimeError: expected scalar type Long but found Float
原因
nn.Linear需要作用于浮点数,这里可能输入了整数类型的张量作为参数。
代码示例
错误版
import torch
import torch.nn as nn
a = torch.tensor([1,2,3,4])
lin = nn.Linear(4,2)
b = lin(a)
print(b)
报错:
文章来源:https://www.toymoban.com/news/detail-699394.html
改正
import torch
import torch.nn as nn
a = torch.tensor([1,2,3,4])
lin = nn.Linear(4,2)
b = lin(a.float())
print(b)
把a转为float,结果为:文章来源地址https://www.toymoban.com/news/detail-699394.html
tensor([-1.1703, 0.0518], grad_fn=<AddBackward0>)
到了这里,关于报错解决:RuntimeError: expected scalar type Long but found Float的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!