如何用numpy保存张量？

Question

我需要通过将 x[-1] 转换为 numpy 矩阵来保存张量，然后将其保存为 pkl 文件。

def forward(self, x):
        
   x=self.forward_features(x)

x[-1] 是什么意思？如何将 x 转换为 numpy ..

Y=[]
Y=x.transform()
With open('result.pkl','w') as file
     Pickle.dump(Y,file)

Answer 1

x[-1] 获取列表中的最后一项。

如果 x 是一个列表，执行 np.array(x) 将列表转换为 numpy 数组。

要将 x 保存到 pickle 文件，请执行

import pickle
import numpy as np

x = np.array([1, 2, 3])

with open('res.pkl', 'wb') as df_file:
    pickle.dump(obj={'x':x}, file = df_file)

How to save tensor with numpy?