张量和张量的区别

Question

我正在使用 Pytorch。 但是，当我使用tensor时出现错误，而使用Tensor时却没有出现任何错误。

张量和张量有什么区别？

这是我的例子：

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

tns.mean()

我收到了这个错误：

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

提前致谢。

Answer 1

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

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

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