如何用numpy保存张量？

【问题标题】：How to save tensor with numpy?如何用numpy保存张量？
【发布时间】：2021-08-24 16:17:39
【问题描述】：

我需要通过将 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)

【问题讨论】：

【解决方案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)

【讨论】：