当前位置: 首页 > 工具软件 > Scalar > 使用案例 >

TypeError: Scalar tensor has no `len()`

郭琦
2023-12-01

我出错的原因是:

利用nn.MSELoss()计算时,输入给的是eagerTensor格式,要求是torch里的tensor格式。

首先我利用了keras调用的vgg预训练模型。

vgg的输入是numpy,输出是eagerTensor格式(与pytorch的tensor格式不一致),所以需要将eagerTensor转为numpy再torch.tensor转为pytorch接受的格式。

 类似资料: