numpy - PyTorch - 到 NumPy 会产生未确定大小的对象？

Question

将 PyTorch 张量转换为 NumPy 我得到

print(nn_result.shape)
# (2433, 2)
np_result = torch.argmax(nn_result).numpy()
type(np_result)
# <type 'numpy.ndarray'>
print(len(np_result))
TypeError: len() of unsized object

为什么？我认为根据文档 numpy()函数将返回一个正确的 ndarray ，但不知何故似乎不完整？

Answer 1

也许你想使用 torch.argmax(nn_result, dim=1) ?自 dim默认为 0，它只返回一个构造为张量的数字。让我用下面的例子来说明:

>>> x = np.array(1)
>>> x.shape
()
>>> len(x)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: len() of unsized object
>>> x = np.array([1])
>>> x.shape
(1,)
>>> len(x)
1

本质上 np.array将占用任何 object您构建的类型。在第一种情况下， object 不是数组，因此您看不到有效的形状。由于它不是调用 len 的数组抛出错误。
torch.argmax与 dim=0返回一个张量，如上例的第一种情况所示，因此返回错误。