为什么Tensor在Pytorch中必须是整数?

时间:2019-06-22 21:16:07

标签: python pytorch

我想得到我的准确性。所以我将正确的数字(张量)除以总数(张量)。然后我得到一个积分而不是浮点数。我想知道为什么。代码如下。另外,如果我将张量传递给numpy,为什么我无法获得确切的数字?我得到像"built-in method numpy of Tensor object at 0x00000296AEDC1168"这样的描述,但我仍然不知道确切的数字。

我尝试了tensor.numpy

print(correct)
tensor(9668)

print(total)
10000

print(correct/total)
tensor(0)

print(100 * correct/total)
tensor(96)

print((100 * correct/total).numpy)
<built-in method numpy of Tensor object at 0x00000296AEDC1168>

1 个答案:

答案 0 :(得分:0)

您可以将.float()correct的{​​{1}}转换为浮动类型。像这样:

total
  

此外,如果我将张量传递给numpy,为什么我无法获得确切的数字?

转换为numpy的正确方法是correct.float()/total 。就您而言:

.numpy()