将GPU的Tensor数据保存至本地查看

step1:将GPU转换为CPU

由于不能对GPU上的tensor直接操作,所以需要先将其转化为CPU数据

# GPU张量->CPU
result = result.cuda().data.cpu().numpy()

step2:保存数据

aaa = sequence_output.data.cpu().numpy()
np.savetxt('result.txt',aaa)

你可能感兴趣的:(将GPU的Tensor数据保存至本地查看)