Función torch.floor de PyTorch


Pytorch torch 参考手册Manual de referencia de PyTorch: torch

torch.floorEs una función de PyTorch que se utiliza para redondear hacia abajo elemento a elemento.

Definición de la función

torch.floor(input, out=None)

Ejemplo de uso

Ejemplo

import torch

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

print(torch.floor(x))

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de PyTorch: torch

Otras extensiones