CODEWITHSUNDEEP
pythonfile-iopytorchtensorfile-writing
Kiran
Kiran

Reputation: 127

Writing tensor to a file in a visually readable manner

In pytorch, I want to write a tensor to a file and visually read the file contents. For example, consider T = torch.tensor([3,4,5,6]). I want to write the tensor T to a file, say file_T.txt, and want to visually read the contents of the file_T.txt, which will be 3,4,5 and 6. How can I achieve this?

Upvotes: 9

Views: 17022

Answers (2)

Tony.sy
Tony.sy

Reputation: 17

You can use numpy.savetxt() but it won't work if your tensor's dimensions are higher than 2. Moreover, this method could be used only if your tensor is on a CPU.

Upvotes: 1

guibs35
guibs35

Reputation: 297

You can use numpy:

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

Upvotes: 13

Related Questions