Python PyTorch floor()用法及代码示例

PyTorch torch.floor()方法返回一个新的张量，该张量是输入元素的下限，小于或等于每个元素的最大整数。
floor method

用法： torch.floor(input, out=None)

参数

input:这是输入张量。
out:输出张量。

返回：它返回张量。

让我们借助几个示例来了解这个概念：
范例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)

输出：

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

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

范例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)

输出：

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

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

相关用法

注：本文由纯净天空筛选整理自PranchalKatiyar大神的英文原创作品 Python – PyTorch floor() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。