Torch.gesv B 应该是二维的。

我刚开始学习牛津机器学习课程,我在使用lua和torch方面还是个新手。

我正在尝试使用torch解决一个简单的线性方程问题。问题是AX = B。

然而,我无法这样做,因为B只是一个1-D张量(向量)。我认为B是向量的情况应该是普遍的。将B复制到2-D张量中是浪费的。

B = torch.Tensor({0.1623, 0.0545})
A = torch.Tensor({{0.1303, -0.0003},{-0.0003, 0.1133}})
X = torch.gesv(B,A)

然后我会得到:

bad argument #2 to '?' (B should be 2 dimensional at
/Users/yiranzhang/torch/pkg/torch/lib/TH/generic/THTensorLapack.c:79)
stack traceback:
[C]: at 0x0af6e890
[C]: in function 'gesv'
[string "xxx = torch.gesv(B,A)"]:1: in main chunk
[C]: in function 'xpcall'
...rs/yiranzhang/torch/install/share/lua/5.1/trepl/init.lua:648: in function 'repl'
...hang/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:185: in main chunk
[C]: at 0x010ae382e0

有什么建议吗?

点赞