Función torch.numel de PyTorch
Manual de referencia de PyTorch torch
torch.numelEs la función de PyTorch utilizada para calcular el número total de elementos de un tensor.
Definición de la función
torch.numel(input)
Ejemplo de uso
Ejemplo
import torch
x = torch.randn(3, 4, 5)
print(Número total de elementos:, torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print(Número total de elementos:, torch.numel(y))
x = torch.randn(3, 4, 5)
print(Número total de elementos:, torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print(Número total de elementos:, torch.numel(y))
El resultado es:
元素总数: 60 元素总数: 5
Otras extensiones