我需要通过将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)
发布于 2021-06-08 01:31:18
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)
https://stackoverflow.com/questions/67874010
复制相似问题