PyTorch torch.fix Fonction
Manuel de référence PyTorch torch
torch.fixest une fonction de PyTorch utilisée pour extraire la partie entière élément par élément (arrondi vers zéro).
Définition de fonction
torch.fix(input, *, out=None)
Exemple d'utilisation
Exemple
import torch
# Exemple d'extraction de la partie entière
x = torch.tensor([-1.5, -0.5, 0.5, 1.5, 2.7])
result = torch.fix(x)
print(result) # tensor([-1., -0., 0., 1., 2.])
# Exemple d'extraction de la partie entière
x = torch.tensor([-1.5, -0.5, 0.5, 1.5, 2.7])
result = torch.fix(x)
print(result) # tensor([-1., -0., 0., 1., 2.])
Autres extensions