已经确定这my_tensor.detach().numpy()是从torch张量获取 numpy 数组的正确方法。
我试图更好地理解为什么。
在刚刚链接的问题的公认答案中,Blupon 指出:
您需要将您的张量转换为另一个除了其实际值定义之外不需要梯度的张量。
在他链接到的第一个讨论中,albanD 指出:
这是预期的行为,因为移动到 numpy 会破坏图形,因此不会计算梯度。
如果您实际上不需要渐变,那么您可以显式地 .detach() 需要 grad 的 Tensor 以获得具有相同内容但不需要 grad 的张量。然后可以将这个其他 Tensor 转换为一个 numpy 数组。
在他链接到的第二个讨论中,apaszke 写道:
变量不能转换为 numpy,因为它们是保存操作历史的张量的包装器,而 numpy 没有这样的对象。您可以使用 .data 属性检索变量持有的张量。然后,这应该有效:var.data.numpy()。
我研究了 PyTorch 的自动分化库的内部工作原理,但我仍然对这些答案感到困惑。为什么它会破坏图形以移动到 numpy?是否因为在 autodiff 图中不会跟踪对 numpy 数组的任何操作?
什么是变量?它与张量有什么关系?
我觉得这里需要一个彻底的高质量 Stack-Overflow 答案,向尚不了解自动分化的 PyTorch 新用户解释原因。
特别是,我认为通过一个图形来说明图形并显示在此示例中断开连接是如何发生的会很有帮助:
Run Code Online (Sandbox Code Playgroud)import torch tensor1 = torch.tensor([1.0,2.0],requires_grad=True) print(tensor1) print(type(tensor1)) tensor1 = tensor1.numpy() print(tensor1) print(type(tensor1))