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

我在Lua中有两个字符串：

a = '01234'
b = '12345'

我想把它们分成几个字符，然后将它们加载到火炬张量中，如下所示：

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)}