Функция torch.numel в PyTorch
Справочное руководство PyTorch torch
torch.numelэто функция в PyTorch, используемая для вычисления общего количества элементов в тензоре.
Определение функции
torch.numel(input)
Примеры использования
Пример
import torch
x = torch.randn(3, 4, 5)
print("Общее количество элементов:", torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print("Общее количество элементов:", torch.numel(y))
x = torch.randn(3, 4, 5)
print("Общее количество элементов:", torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print("Общее количество элементов:", torch.numel(y))
Результат вывода:
元素总数: 60 元素总数: 5
Другие расширения