以视觉上可读的方式将张量写入文件

Question

在pytorch中，我想将张量写入文件并直观地读取文件内容。例如，考虑 T = torch.tensor([3,4,5,6](。我想将张量 T 写入文件，比如file_T.txt，并希望直观地读取file_T.txt的内容，即 3、4、5 和 6。我怎样才能做到这一点？

Answer 1

您可以使用 numpy：

import numpy as np
np.savetxt('my_file.txt', torch.Tensor([3,4,5,6]).numpy())

Answer 2

您可以使用numpy.savetxt()但如果张量的维数大于 2，则它不起作用。此外，仅当您的张量位于 CPU 上时，才能使用此方法。