python - 张量和张量的区别

Question

我正在使用 Pytorch。但是当我使用时tensor出现错误，而使用Tensor.

张量和张量有什么区别？

这是我的例子：

tns = torch.tensor([91,21,34,56])

tns.mean()

我收到了这个错误：

RuntimeError：只能计算浮点类型的平均值。反而得到了长。

提前致谢。

标签： pythonpython-3.xpytorch

Answer 1

如果您未在其中指定 a dtype，torch.tensor()则从数据中推断出这一点。由于您的数据都是整数，因此它使用了Long.

torch.mean()只能计算浮点数。

默认情况下，torch.Tensor它是一个别名，torch.FloatTensor因此它自动具有 dtype float（除非您更改默认行为）。