在 numpy 和 pytorch 中 1.== 1.0 吗？

Question

等方法的输出

a=torch.ones(1,4)

是

tensor([[1., 1., 1., 1.]])

数字是“1”吗？是“1.0”的缩写，还是与整数“1”完全相同？

很抱歉问这么简单的问题，因为我不知道哪些关键词适合在官方文档中搜索。

Answer 1

在 PyTorch 中，数字 '

1.

' 相当于 '

1.0

'，它是浮点数。

所以，它与整数“

1

”不同。在您提供的张量中，数字表示为带有一位小数的浮点数，而不是整数。