lua - 如何向 Torch7 张量添加切片？

Question

我正在使用 Lua 和 Torch 7。

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

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

标签： luatorch

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