PyTorch torch.numel-Funktion
PyTorch torch-Referenzhandbuch
torch.numelist eine Funktion in PyTorch, mit der die Gesamtzahl der Elemente in einem Tensor berechnet wird.
Funktionsdefinition
torch.numel(input)
Anwendungsbeispiele
Beispiel
import torch
x = torch.randn(3, 4, 5)
print("Gesamtzahl der Elemente:", torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print("Gesamtzahl der Elemente:", torch.numel(y))
x = torch.randn(3, 4, 5)
print("Gesamtzahl der Elemente:", torch.numel(x))
y = torch.tensor([1, 2, 3, 4, 5])
print("Gesamtzahl der Elemente:", torch.numel(y))
Das Ausgabeergebnis ist:
元素总数: 60 元素总数: 5
Andere Erweiterungen