python - RuntimeError('dot : 期望两个向量具有相同的 dtype，但发现 Double 和 Float

Question

我正在尝试执行此操作：

L_c += 1 - torch.abs(torch.dot(nuNormalized, Nu))

在哪里：

numNormalized -> tensor([ 0.3790, -0.4208, 0.8242], dtype=torch.float64)

Nu -> tensor([-0.9961, -0.9961, -0.9961], device='cuda:0')

我收到此错误：

(<class 'RuntimeError'>, RuntimeError('dot : 期望两个向量具有相同的 dtype，但发现 Double 和 Float',), <traceback object at 0x7f4d276155c8>)

请问有什么建议吗？谢谢！

标签： pythonpycharmpytorch

Answer 1

将 nuNormalized 转换为从 double 浮动。

您也可以考虑将两者移至同一设备。