使用 numpy 的 savez_compressed 存储深度学习模型生成的向量数据

发布于：2023-03-07 ⋅ 阅读:(516) ⋅ 点赞:(0)

import numpy
from numpy.lib.npyio import NpzFile

numpy.savez_compressed('nn.npz', vectors=[0.11, 0.22])


data: NpzFile = numpy.load('nn.npz')

print(list(data.keys()))

v: numpy.ndarray = data['vectors']

print(v)
print(type(v))