当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


PyTorch torch.fmod()方法给出除数除以元素的余数。除数可以是数字或张量。

用法: torch.fmod(input, div, out=None)

参数

  • input:这是输入张量。
  • div:这可以是数字或张量。
  • out:输出张量。

返回:它返回张量。

让我们借助几个示例来了解这个概念:
范例1:

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([5, 6, 7, 4]) 
print(a) 
  
# Applying the fmod function and  
# storing the result in 'out' 
out = torch.fmod(a, 3) 
print(out)

输出:

5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 0
 1
 1
[torch.FloatTensor of size 4]

范例2:

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([5, 6, 7, 4]) 
b = torch.FloatTensor([2, 3, 4, 1]) 
print(a, b) 
  
# Applying the fmod function and  
# storing the result in 'out' 
out = torch.fmod(a, b) 
print(out)

输出:

5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 3
 4
 1
[torch.FloatTensor of size 4]

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

相关用法


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