首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何用numpy保存张量?

如何用numpy保存张量?
EN

Stack Overflow用户
提问于 2021-06-07 22:57:45
回答 1查看 104关注 0票数 0

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

代码语言:javascript
运行
复制
def forward(self, x):
        
   x=self.forward_features(x)

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

代码语言:javascript
运行
复制
Y=[]
Y=x.transform()
With open('result.pkl','w') as file
     Pickle.dump(Y,file)
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-06-08 01:31:18

x[-1]将为您提供列表中的最后一项。

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

要将x保存到pickle文件,请执行以下操作

代码语言:javascript
运行
复制
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)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67874010

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档