張量和張量的區別

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 浮點數（除非您更改默認行為）。