我正在尝试更新Torch中矩阵中的一个段,但我没有一个简单的功能来执行它。
例如:$ x = torch.Tensor(5, 6):zero() -- create a matrix
我尝试更新矩阵中的段,如:
$ x[{ 2,{2,4} }]= {1, 2, 3}
和
$ x[{ 2,{2,4} }]:fill({1,2,3})
它们都不起作用。
我应该只使用循环还是存在一个简单的函数?
答案 0 :(得分:0)
只需创建张量而不是尝试直接分配表格:
x[{2, {2,4}}] = torch.Tensor({1, 2, 3})