pytorch .detach().cpu().numpy()

深度学习模型用GPU训练数据时，需要将数据转换成tensor类型，输出也是tensor类型。

detach():

返回一个new tensor，但是不再有梯度。
如果想要将cuda tensor类型数据转成numpy，需要先将其转换成cpu float-tensor类型， 再转成numpy。

标签：tensor,pytorch,类型,detach,numpy,cpu
来源： https://blog.csdn.net/qq_42822946/article/details/119085497