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

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

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

detach():

返回一个new tensor,但是不再有梯度。
如果想要将cuda tensor类型数据转成numpy,需要先将其转换成cpu float-tensor类型, 再转成numpy。
上一篇:C++创建线程


下一篇:linux中pthread_join()与pthread_detach()详解