在pytorch中,我想将张量写入文件并直观地读取文件内容。例如,考虑 T = torch.tensor([3,4,5,6])。我想将张量 T 写入一个文件,比如 file_T.txt,并且想直观地读取 file_T.txt 的内容,即 3、4、5 和 6。我该如何实现?
最佳答案
您可以使用 numpy:
import numpy as np
np.savetxt('my_file.txt', torch.Tensor([3,4,5,6]).numpy())
关于python - 以视觉可读的方式将张量写入文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62412976/