0
点赞
收藏
分享

微信扫一扫

报错解决:RuntimeError: expected scalar type Long but found Float

文章目录

报错信息

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)

报错:
在这里插入图片描述

改正

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,结果为:

tensor([-1.1703,  0.0518], grad_fn=<AddBackward0>)
举报

相关推荐

0 条评论