如何向 Torch7 张量添加切片？

Question

我正在使用 Lua 和 Torch 7。

如何有效地（不使用 for 循环）将大小为 (1, 2048, 1024) 的张量 A 和 B 相加以生成大小为 (2, 2048, 1024) 的张量 C？

我正在使用https://github.com/torch/torch7/blob/master/doc/tensor.md 的软件包和文档，但我没有看到可以帮助我的功能。

Answer 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