Python – PyTorch fmod() 方法

在本文中，我们将了解 PyTorch 中的 fmod() 方法的使用。fmod() 方法用于计算输入张量中元素的元素级余数。该方法类似于NumPy中的fmod函数。

语法

torch.fmod(input, divisor, *, out=None) → Tensor

参数

• input(Tensor)：输入张量。
• divisor(Tensor或标量)：除数。
• out(Tensor, optional)：输出张量。

返回值

计算输入张量中的元素级余数并返回张量。返回的张量与输入张量具有相同的形状。

示例

import torch

# 两个张量的余数计算
x = torch.tensor([3, 4, 5])
y = torch.tensor([2, 3, 4])
z = torch.fmod(x, y)

print(z)  # tensor([1, 1, 1])

上面的示例中，我们首先创建了两个张量x和y，然后将它们作为参数传递给了fmod()方法。fmod()方法计算了x中每个元素对y求余的结果，并将其保存在新的张量z中。

注意事项

• 当除数是一个标量时，fmod()方法相当于对输入张量中的每个元素对该标量求余。
• 当除数是一个张量时，fmod()方法会使该张量与输入张量具有相同的形状，并对每个相应的元素对其索引的除数张量求余。

这是PyTorch中的fmod()方法的用法及示例。希望本文能够帮助您更好地了解fmod()方法及其用法。