Как добавить ломтик в Torch7 Tensor? - PullRequest
0 голосов
/ 16 мая 2018

Я использую Lua и Torch 7.

Как эффективно (без петель) сложить вместе тензоры A и B размера (1, 2048, 1024), чтобы получить тензор C размера (2, 2048, 1024)?

Я использую пакет и документацию на https://github.com/torch/torch7/blob/master/doc/tensor.md, но не вижу функции, которая может мне помочь.

1 Ответ

0 голосов
/ 16 мая 2018

Некоторые функции находятся в библиотеке 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

...