cpu tensor转numpy:

#假定a为tensor

a.numpy()

gpu tensor转numpy:

gpu下的tensor不能直接转numpy,需要先转到cpu tensor后再转为numpy

a.cpu().numpy()

注:若tensor带有梯度,以上述方式转换时会报错:

RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.

可以改用detach().numpy()方式

a.detach().numpy()
a.cpu().detach().numpy()

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐