当前位置 : 主页 > 网络编程 > lua >

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

来源:互联网 收集:自由互联 发布时间:2021-06-23
如果我有一个张量: t1 = torch.Tensor(2, 2) 有没有办法将这些数据作为Lua表获取? create a tensor from a table有一个专用的构造函数,但到目前为止还没有开箱即用的方法来转换相反的方法. 当然
如果我有一个张量:

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)转换器.

网友评论