RuntimeError: Can‘t call numpy() on Tensor that requires grad

description¶

pytorch Tensor to numpy got error:

RuntimeError: Can‘t call numpy() on Tensor that requires grad

solution¶

Use Tensor.detach().numpy() when converting to numpy:

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

When the tensor in the calculation is converted, because it has a gradient value, it cannot be directly converted to numpy format, so it is best to call .detach().numpy() anyway.