pytorch Tensor转numpy并解决RuntimeError: Can‘t call numpy() on Tensor that requires grad.报错

解决方法

转numpy时使用Tensor.detach().numpy()：

a = torch.ones(5)
b = a.detach().numpy()
print(b)

问题解析

当计算中的tensor转换时，由于它带梯度值时，因此不能直接转为numpy格式，所以最好不论如何都调用一下.detach().numpy()