将 Torch Lua 转换为 Numpy Python

Question

我希望将一些 Lua Torch 代码转换为 numpy Python。 我用谷歌搜索了一些文档，但仍然不清楚。

我找到了这个。 https://nn.readthedocs.io/en/rtd/index.html

想知道 Lua Torch 函数和 numpy 函数之间是否有任何映射？

谢谢

Answer 1

您可以尝试Lutorpy，这是一个用于将 lua/torch 与 python/numpy 连接起来的 python 库。

这是一个转换的例子：

-- lua code                             # python code (with lutorpy)
--                                      import lutorpy as lua
require "nn"                    ===>    require("nn")
model = nn.Sequential()         ===>    model = nn.Sequential()
-- use ":" to access add        ===>    # use "._" to access add
model:add(nn.Linear(10, 3))     ===>    model._add(nn.Linear(10, 3))
--                                      import numpy as np
x = torch.Tensor(10):zero()     ===>    arr = np.zeros(10)
-- torch style(painful?)        ===>    # numpy style(elegent?) 
x:narrow(1, 2, 6):fill(1)       ===>    arr[1:7] = 1
--                                      # convert numpy array to a torch tensor
--                                      x = torch.fromNumpyArray(arr)
--                                      # or you can still use torch style
x:narrow(1, 7, 2):fill(2)       ===>    x._narrow(1, 7, 2)._fill(2)
-- 1-based index                ===>    # 0-based index
x[10] = 3                       ===>    x[9] = 3
y = model:forward(x)            ===>    y = model._forward(x)
--                                      # you can convert y to a numpy array
--                                      yArr = y.asNumpyArray()

更多信息，您可以前往lutorpy的github页面。