Files
candle/candle-core/tests/npy.py
2023-10-12 15:25:05 +02:00

10 lines
193 B
Python

import numpy as np
x = np.arange(10)
# Write a npy file.
np.save("test.npy", x)
# Write multiple values to a npz file.
values = { "x": x, "x_plus_one": x + 1 }
np.savez("test.npz", **values)