Fonction torch.floor de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.floorest une fonction de PyTorch utilisée pour l'arrondi vers le bas élément par élément.

Définition de fonction

torch.floor(input, out=None)

Exemple d'utilisation

Exemple

import torch

x = torch.tensor([1.1, 2.5, 3.7, -1.2])

print(torch.floor(x))

Le résultat de sortie est:

tensor([1., 2., 3., -2.])

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions