0

我刚开始参加牛津机器学习课程,而且我是 lua 和 torch 的新手。

我正在尝试用火炬解决一个简单的线性方程问题。问题就像 AX = B

但是,我无法做到这一点,因为 B 只是一维张量(向量)。我认为 B 是向量的情况应该是常见的。将 B 复制到二维张量是浪费的。

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

有什么建议么?

4

0 回答 0