# model_out为CUDA上的tensor
model_out = model_out.cpu()
# detach():去除梯度
model_out = model_out.detach().numpy()
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/180258.html原文链接:https://javaforall.net
Pytorch cuda上的tensor转numpy[通俗易懂]#model_out为CUDA上的tensormodel_out=model_out.cpu()#detach():去除梯度model_out=model_out.detach().numpy()
# model_out为CUDA上的tensor
model_out = model_out.cpu()
# detach():去除梯度
model_out = model_out.detach().numpy()
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/180258.html原文链接:https://javaforall.net