pytorch报错RuntimeError: expected device cpu and dtype Float but got device cpu and dtype Long

最新推荐文章于 2023-11-08 22:49:00 发布

原创最新推荐文章于 2023-11-08 22:49:00 发布 · 3.7k 阅读

0 ·

CC 4.0 BY-SA版权

本文解决了一个常见的PyTorch编程问题，当尝试使用整数类型的SCORES与浮点数类型的prob进行操作时，会遇到类型不匹配的错误。通过将SCORES转换为浮点数类型，可以避免此类运行时错误。

部署运行你感兴趣的模型镜像

prob * SCORES 报错：
RuntimeError: expected device cpu and dtype Float but got device cpu and dtype Long
其中prob和SCORES的数据及类型是：
SCORES = torch.tensor([[1, 2, 3, 4, 5]])
In [71]: prob
Out[71]:
tensor([[0.2000, 0.2000, 0.2000, 0.2000, 0.2000],
        [0.2001, 0.2002, 0.1998, 0.2001, 0.1998],
        [0.2000, 0.2000, 0.2000, 0.2000, 0.2000],
        ...,
        [0.2000, 0.2000, 0.2000, 0.2000, 0.2000],
        [0.2000, 0.2000, 0.2000, 0.2000, 0.2000],
        [0.1999, 0.2000, 0.2000, 0.2000, 0.2000]], grad_fn=<SoftmaxBackward>)

解决方法：
将SCORES由整数类型改为float类型：
SCORES = torch.tensor([[1., 2., 3., 4., 5.]])
或者：
SCORES = SCORES.float()