Python – PyTorch frac() 方法

PyTorch torch.frac()方法计算输入中每个元素的小数部分。

Syntax: torch.frac(input, out=None)

Arguments

• input: This is input tensor.
• out: The output tensor.

Return: It returns a Tensor.

让我们通过几个例子来看看这个概念：
示例 1：

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.FloatTensor([-5.4, 6.6, -7.1099, 4.4567])
print(a)
  
# Applying the frac function and 
# storing the result in 'out'
out = torch.frac(a)
print(out)

输出：

-5.4000
 6.6000
-7.1099
 4.4567
[torch.FloatTensor of size 4]

-0.4000
 0.6000
-0.1099
 0.4567
[torch.FloatTensor of size 4]

示例 2：

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.randn(6)
print(a)
  
# Applying the frac function and 
# storing the result in 'out'
out = torch.frac(a)
print(out)

输出：

-0.5260
-1.8843
 0.8062
 1.2264
-0.1505
-0.3217
[torch.FloatTensor of size 6]

-0.5260
-0.8843
 0.8062
 0.2264
-0.1505
-0.3217
[torch.FloatTensor of size 6]

在评论中写代码？请使用 ide.geeksforgeeks.org，生成链接并在此处分享链接。