CODEWITHSUNDEEP
numpytensor
Ifeoluwa Olubo
Ifeoluwa Olubo

Reputation: 39

How do i save a numpy tensor to a file

I am trying to convert images to numpy tensor that is 4 dimensional and I want to save this into a file, preferrably a csv file. I already have the image array in the dimension I want them but now the problem is saving to a file. Please how do I go about doing this.

Upvotes: 2

Views: 11626

Answers (1)

Pascal Getreuer
Pascal Getreuer

Reputation: 3271

Numpy has a lot of options for IO of array data:

  • If binary format is Ok, you can use np.save to save the 4D tensor in a binary (".npy") format. The file can be read again with np.load. This is a very convenient way to save numpy data, and it works for numeric arrays of any number of dimensions.

  • np.savetxt can write a 1D or 2D array in CSV-like text format. You could use np.reshape to flatten your tensor down to 1D or 2D and then use np.savetxt. The downside is the file doesn't keep track of the full 4D shape, so you'll need to track that separately in some way.

  • If storing in text representation is important, a better option may be to convert the tensor to string with np.array2string, then write the string to file. This works even for arrays with more than 2 dimensions.

  • The .tofile method simply dumps the element data as a raw binary file. No shape or another other metadata is preserved, but the binary file is easy to read into other programs.

Upvotes: 8

Related Questions