将字符串字符拆分为火炬张量

Splitting String Characters into a Torch Tensor

我在 Lua 中有 2 个字符串:

a = '01234'
b = '12345'

我想将它们拆分成字符并将它们加载到 Torch Tensor 中,如下所示:

tens = torch.Tensor{{0, 1, 2, 3, 4}, {1, 2, 3, 4, 5}} 

谢谢。

function split(str)
    tbl = {}
    for c in str:gmatch('.') do
         table.insert(tbl, c)
    end
    return tbl
end

a = '01234'
b = '12345'

tens = torch.Tensor{split(a), split(b)}