我正在使用Lua和Torch 7.
我如何有效地(没有for循环)将尺寸(1,2048,1024)的张量A和B加在一起以产生尺寸为(2,2048,1024)的Tensor C?
我正在使用https://github.com/torch/torch7/blob/master/doc/tensor.md上的软件包和文档,但我没有看到可以帮助我的功能。
答案 0 :(得分:1)
某些功能位于torch
库而不是tensor
https://github.com/torch/torch7/blob/master/doc/maths.md#res-torchcat-res-x_1-x_2-dimension-。
在连接之前,可能需要此函数来添加第三维:
https://github.com/torch/torch7/blob/master/doc/tensor.md#result-expandresult-sizes