Я использую Arrayfire и Flashlight для оценки сети.
auto tmp = output(af::seq(2, 10), af::span, af::span, af::span);
auto softmax_tmp = fl::softmax(tmp, 0);
output(af::seq(2,10),af::span,af::span,af::span)=softmax_tmp;
output - тензор с формой (12,100,1,1). Теперь я хочу вытащить (2,10) диммера тензора и для извлеченных 100 9-дим-векторов применить к ним операцию softmax. Тогда положи их обратно. Коды выше.
Проблема в том, что 3-я строка не работает. softmax_tmp - правильное значение, но оператор присваивания в 3-й строке просто не удался. Именно он может успешно пройти компиляцию, но output остается старым значением, как в 1-й строке.
Кто мог мне помочь? Большое спасибо на самом деле.