Python Pyrotch dot产品在某些情况下不起作用

我正在努力解决一个我无法用PyTorch代码解决的问题。我的任务很简单：得到两个张量的内积。但是，在某些情况下，输出变为零，而不是返回正确的值

例如，下面的代码演示了这个问题

a = torch.tensor([3,1])
b = torch.tensor([3.,1.])
a.dot(a)  # this returns 10, which is correct
b.dot(b)  # this returns 0., which is not correct

我正在使用PyTorch 1.2版，这在我之前从未发生过。。。在我定义张量的方法中，有没有一些简单的东西我遗漏了

---

谢谢你抽出时间

我刚刚使用torch版本运行了相同的代码

'1.0.1.post2'

，并分别看到了

a.dot（a）

和

b.dot（b）

的结果。请确认代码中是否缺少其他内容。

您的代码在torch 1.1.0 Hanks中运行良好！我发现将Python3.6.3升级到Python3.6.6就可以了。我不知道为什么会这样，但现在已经解决了。谢谢！根据您的评论，我认为代码很好，所以我尝试将Python 3.6.3升级到Python 3.6.6，解决了这个问题。