在pytorch中，我想把一个Tensor写入一个文件，然后从视觉上读取文件内容。例如，假设T = torch.tensor（[3，4，5，6]）。我想把TensorT写入一个文件，比如file_T. txt，然后从视觉上读取file_T. txt的内容，分别是3、4、5和6。我该如何实现这一点？
9jyewag01#
您可以使用numpy：
import numpy as np np.savetxt('my_file.txt', torch.Tensor([3,4,5,6]).numpy())
mbyulnm02#
你可以使用numpy.savetxt()，但是如果你的Tensor的维数大于2，它就不起作用。而且，这种方法只能在你的Tensor在CPU上时使用。
numpy.savetxt()
2条答案
按热度按时间9jyewag01#
您可以使用numpy：
mbyulnm02#
你可以使用
numpy.savetxt()，但是如果你的Tensor的维数大于2，它就不起作用。而且，这种方法只能在你的Tensor在CPU上时使用。