PyTorch torch.round Funktion
PyTorch torch Referenzhandbuch
torch.roundEs ist eine Funktion in PyTorch für elementweises Runden.
Funktionsdefinition
torch.round(input, decimals, out)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([0.1, 0.5, 0.9, 1.2, 1.5])
print(torch.round(x))
# Geben Sie die Anzahl der Dezimalstellen an
y = torch.tensor([1.234, 5.678])
print(torch.round(y, decimals=2))
x = torch.tensor([0.1, 0.5, 0.9, 1.2, 1.5])
print(torch.round(x))
# Geben Sie die Anzahl der Dezimalstellen an
y = torch.tensor([1.234, 5.678])
print(torch.round(y, decimals=2))
Das Ausgabeergebnis ist:
tensor([0., 0., 1., 1., 2.]) tensor([1.2300, 5.6800])
Andere Erweiterungen