PyTorch 中遇到的问题

米饭的白色

已于 2022-07-21 16:12:49 修改

阅读量575

点赞数

CC 4.0 BY-SA版权

分类专栏： Python fealpy PyTorch 文章标签： pytorch

于 2022-07-20 19:06:03 首次发布

本文链接：https://blog.csdn.net/mifangdebaise/article/details/125898427

Python 同时被 3 个专栏收录

56 篇文章

订阅专栏

fealpy

16 篇文章

订阅专栏

PyTorch

8 篇文章

订阅专栏

在使用PyTorch的einsum函数时遇到错误，提示输入张量数据类型不一致。经过检查，发现张量c_ws为DoubleTensor，u_minus_c为FloatTensor。为了解决这个问题，可以将其中一个张量转换为相同的数据类型，例如将u_minus_c转换为DoubleTensor或者将c_ws转换为FloatTensor。这样做可以确保所有输入张量的数据类型一致，从而避免运行时错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

调用 torch.einsum('i, ijk, j->jk', c_ws, u_minus_c, trainCellVol) 时, 提示了如下错误

  File "/Users/yczhang/opt/anaconda3/envs/fealpy/lib/python3.8/site-packages/torch/functional.py", line 241, in einsum
    return torch._C._VariableFunctions.einsum(equation, operands)
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2' in call to _th_mm_out

根据错误的提示, 可能是 c_ws, u_minus_c, trainCellVol 的数据类型不一致造成的.
并注意, pytorch 中的 Tensor 数据 float64 是 double ，而 float32 是标准的 float.
然后查看下各数据的类型:

c_ws.type()
'torch.DoubleTensor'

u_minus_c.type()
'torch.FloatTensor'

trainCellVol.type()
'torch.DoubleTensor'

那么, 将 c_ws.float(), 或将 u_minus_c.double() 是都可以的, 即下面的两种方式:

a0 = torch.einsum('i, ijk, j->jk', c_ws, u_minus_c.double(), trainCellVol) 
a0 = torch.einsum('i, ijk, j->jk', c_ws.float(), u_minus_c, trainCellVol)

pytorch 查看模型结构网络参数