Сводка по оси в PyTorch

У метода sum сводка может идти не по всем ячейкам, а вдоль одной оси. Ось с номером 0 суммирует строки. Ось с номером 1 суммирует столбцы.

Возьмём таблицу из двух строк и трёх столбцов. Сложим по первой оси и выведем результат вместе с его формой:

import torch

table = torch.tensor([[1, 2, 3], [4, 5, 6]])
by_cols = table.sum(dim=0)
print(by_cols) # выведет tensor([5, 7, 9])
print(by_cols.shape) # выведет torch.Size([3])

Теперь сложим по второй оси и снова посмотрим форму:

import torch

table = torch.tensor([[1, 2, 3], [4, 5, 6]])
by_rows = table.sum(dim=1)
print(by_rows) # выведет tensor([ 6, 15])
print(by_rows.shape) # выведет torch.Size([2])

Для таблицы цен [[10, 20], [30, 40]] сложите по столбцам и выведите ряд сумм и его форму.

Для таблицы баллов [[1, 2, 3], [4, 5, 6]] сложите по строкам и выведите ряд сумм и его форму.

Для таблицы отметок [[2, 0], [1, 3], [4, 1]] сложите по столбцам и выведите ряд сумм.