加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 大数据 > 正文

如何将火炬Tensor / Storage转换为lua表?

发布时间:2020-12-15 00:24:13 所属栏目:大数据 来源:网络整理
导读:如果我有一个张量: t1 = torch.Tensor(2,2) 有没有办法将这些数据作为Lua表获取? 解决方法 create a tensor from a table有一个专用的构造函数,但到目前为止还没有开箱即用的方法来转换相反的方法. 当然你可以手动完成: -- This assumes `t1` is a 2-dime
如果我有一个张量:
t1 = torch.Tensor(2,2)

有没有办法将这些数据作为Lua表获取?

解决方法

create a tensor from a table有一个专用的构造函数,但到目前为止还没有开箱即用的方法来转换相反的方法.

当然你可以手动完成:

-- This assumes `t1` is a 2-dimensional tensor!
local t2 = {}
for i=1,t1:size(1) do
  t2[i] = {}
  for j=1,t1:size(2) do
    t2[i][j] = t1[i][j]
  end
end

更新:截至commit 10f3323,现在有一个专用的torch.totable(object)转换器.

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读