pytorch .detach().cpu().numpy()2024-04-11 14:41:55 pytorch .detach().cpu().numpy() 深度学习模型用GPU训练数据时,需要将数据转换成tensor类型,输出也是tensor类型。 detach(): 返回一个new tensor,但是不再有梯度。 如果想要将cuda tensor类型数据转成numpy,需要先将其转换成cpu float-tensor类型, 再转成numpy。 上一篇:C++创建线程下一篇:linux中pthread_join()与pthread_detach()详解