Метод nelement
Метод nelement возвращает общее количество элементов,
содержащихся в тензоре. Это произведение всех размерностей
тензора. Метод не принимает никаких параметров и всегда
возвращает целое число типа int. Данный метод является
синонимом метода numel и полностью идентичен ему
по функциональности.
Синтаксис
tensor.nelement()
Пример
Давайте создадим одномерный тензор из пяти элементов
и узнаем общее количество элементов с помощью метода
nelement:
import torch
t = torch.tensor([1, 2, 3, 4, 5])
res = t.nelement()
print(res)
Результат выполнения кода:
5
Пример
Рассмотрим двумерный тензор размером 2x3:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.nelement()
print(res)
Результат выполнения кода:
6
Общее количество элементов равно произведению размерностей: 2 × 3 = 6.
Пример
Метод nelement корректно работает с тензорами
любой размерности, включая трёхмерные. Создадим тензор
размером 2x2x2 и проверим количество элементов:
import torch
t = torch.tensor([
[
[1, 2],
[3, 4],
],
[
[5, 6],
[7, 8],
],
])
res = t.nelement()
print(res)
Результат выполнения кода:
8
Пример
Метод nelement возвращает 1 для тензора
с одним элементом (скалярного тензора). Создадим
скалярный тензор:
import torch
t = torch.tensor(42)
res = t.nelement()
print(res)
Результат выполнения кода:
1
Даже если тензор содержит всего одно число, метод возвращает 1.
