python - numpy 数据存储在哪里？

Question

在python中，如果只导入torch（而不是导入numpy），“torch.numpy()”仍然可以工作。这是否意味着可以在没有 numpy 包的情况下存储和显示 numpy 数据？numpy 数据存储在哪里以及如何显示（没有 numpy 包）？

示例代码：

import torch
a = torch.tensor([[1,2,3],[4,5,6]])
a = a.numpy()
print(a)
array([[1, 2, 3],
       [4, 5, 6]])

标签： pythonnumpypytorch

Answer 1

PyTorch 在内部使用 NumPy。您不需要手动导入包使用的所有内容，这是模块的核心原则之一。它仍然是同一个 NumPy 类的对象，您需要安装 NumPy 才能使其工作，否则您会收到导入错误，只是导入发生在 PyTorch 的一个文件中，而不是您自己的文件中。