PyTorch torch.is_floating_point Funktion
PyTorch torch-Referenzhandbuch
torch.is_floating_pointEs ist eine Funktion in PyTorch, mit der überprüft wird, ob der Datentyp eines Tensors ein Gleitkommatyp ist.
Funktionsdefinition
torch.is_floating_point(input)
Anwendungsbeispiel
Beispiel
import torch
# Gleitkomma-Tensor
x = torch.tensor([1.0, 2.0, 3.0])
print("Gleitkomma:", torch.is_floating_point(x))
# Ganzzahliger Tensor
y = torch.tensor([1, 2, 3])
print("Ganzzahl:", torch.is_floating_point(y))
# Gleitkomma-Tensor
x = torch.tensor([1.0, 2.0, 3.0])
print("Gleitkomma:", torch.is_floating_point(x))
# Ganzzahliger Tensor
y = torch.tensor([1, 2, 3])
print("Ganzzahl:", torch.is_floating_point(y))
Das Ausgabeergebnis ist:
浮点数: True 整数: False
Andere Erweiterungen