我的损失是Tensor列表如下：[tensor(0.0153, device='cuda:0', grad_fn=<DivBackward0>), tensor(0.0020, device='cuda:0', grad_fn=<DivBackward0>)]如何将其转换为NumPy数组？非常感谢！
[tensor(0.0153, device='cuda:0', grad_fn=<DivBackward0>), tensor(0.0020, device='cuda:0', grad_fn=<DivBackward0>)]
axr492tv1#
一个Tensor包含了比它的值更多的信息，例如关于它的反向传播梯度的信息。Tensor的item属性隔离了它的值。假设loss是我们的损失列表，要将它作为一个numpy数组，我们可以执行以下操作：
Tensor
item
loss
losses_np = np.array([x.item() for x in losses])
对于类似的问题，Tensor的values和numpy属性也可能有用。
values
numpy
1条答案
按热度按时间axr492tv1#
一个
Tensor包含了比它的值更多的信息，例如关于它的反向传播梯度的信息。Tensor的item属性隔离了它的值。假设loss是我们的损失列表，要将它作为一个numpy数组，我们可以执行以下操作：对于类似的问题，Tensor的
values和numpy属性也可能有用。