Число элементов в PyTorch

Число элементов - это сколько всего значений лежит в тензоре. Форма описывает раскладку по осям, а не общий счёт. Длина списка Python тоже не подходит: у таблицы len даёт число рядов, а не все ячейки сразу.

Общий счёт возвращает метод numel. Создадим ряд из четырёх чисел и выведем число элементов:

import torch

scores = torch.tensor([1, 2, 3, 4])
print(scores.numel()) # выведет 4

Если в тензоре одна ячейка, метод numel вернёт 1:

import torch

price = torch.tensor([99])
print(price.numel()) # выведет 1

У таблицы элементы перемножают по осям. Ряд из двух строк по три числа содержит 6 значений:

import torch

grid = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(grid.numel()) # выведет 6

Соберите ряд из чисел 11, 22 и 33 и выведите, сколько в нём элементов всего.

Соберите таблицу из двух рядов по четыре числа и выведите общее число её ячеек.

Соберите ряд из шести целых чисел и выведите, сколько значений в нём хранится.