Сумма элементов в PyTorch

Метод sum складывает элементы тензора. Без указания оси он возвращает одно число - сумму всех ячеек сразу.

Создадим одномерный ряд и выведем сумму всех его элементов:

import torch

scores = torch.tensor([4, 5, 3])
total = scores.sum()
print(total) # выведет tensor(12)

Тот же метод складывает все ячейки таблицы, не различая строки и столбцы:

import torch

table = torch.tensor([[1, 2, 3], [4, 5, 6]])
total = table.sum()
print(total) # выведет tensor(21)

Соберите ряд цен 2, 3 и 5 и выведите их сумму одним числом.

Соберите таблицу баллов [[1, 1], [1, 1]] и выведите сумму всех ячеек.

Соберите ряд отметок 10, 20, 30 и 40 и выведите сумму всех элементов.