# model_out为CUDA上的tensor model_out = model_out.cpu() # detach():去除梯度 model_out = model_out.detach().numpy()