【无标题】

tensor保存到txt

取data后转Numpy。因为我的输出是三维所以reshape为二维。

np.savetxt('data.txt',  enc_outputs.data.numpy().reshape(-1,1))

你可能感兴趣的:(numpy,python)