PyTorch torch.floor() method returns a new tensor which is floor of the elements of input, the largest integer less than or equal to each element.
Syntax:
torch.floor(input, out=None)Arguments
- input: This is input tensor.
- out: The output tensor.
Return: It returns a Tensor.
Example 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) |
Output:
1.5000 4.2000 6.7000 3.9000 [torch.FloatTensor of size 4] 1 4 6 3 [torch.FloatTensor of size 4]
Example 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) |
Output:
1.5900 4.5999 6.7800 4.0000 [torch.FloatTensor of size 4] 1 4 6 3 [torch.FloatTensor of size 4]
<!–
–>











Please Login to comment…