Python – método PyTorch floor()

El método PyTorch torch.floor()devuelve un nuevo tensor que es el piso de los elementos de entrada, el entero más grande menor o igual a cada elemento.

Sintaxis: torch.floor(input, out=None)

Argumentos

• entrada: Este es el tensor de entrada.
• out: El tensor de salida.

Return: Devuelve un Tensor.

Veamos este concepto con la ayuda de algunos ejemplos:
Ejemplo 1:

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.FloatTensor([1.5, 4.2, 6.7, 3.9])
print(a)
  
# Applying the floor function and 
# storing the result in 'out'
out = torch.floor(a)
print(out)

Producción:

 1.5000
 4.2000
 6.7000
 3.9000
[torch.FloatTensor of size 4]

 1
 4
 6
 3
[torch.FloatTensor of size 4]

Ejemplo 2:

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.FloatTensor([1.59, 4.5999, 6.78, 3.99999])
print(a)
  
# Applying the floor function and 
# storing the result in 'out'
out = torch.floor(a)
print(out)

Producción:

1.5900
 4.5999
 6.7800
 4.0000
[torch.FloatTensor of size 4]

 1
 4
 6
 3
[torch.FloatTensor of size 4]