Сумма столбца в факеле

Question

Как суммировать колонку в факеле? У меня тензор 128 * 1024, и я хочу получить тензор 1 * 1024, суммируя все строки.

Например: : с

1 2 3 4 5 6

Я хочу б

5 7 9

Answer 1

Чтобы сделать это, вы можете использовать метод суммы.

torch.sum(a,1)

В общем, вы можете указать любую ось вы хотите просуммировать.

torch.sum(a,axis)

(Суммируя по строкам, вы можете использовать ось = 2)