Python PyTorch 余数()方法

在PyTorch中，torch.remainder() 方法用于计算两个张量的元素的余数。

语法

以下是 remainder() 方法的语法：

torch.remainder(input, other, *, out=None) → Tensor

参数

remainder() 方法接受以下参数：

• input：张量。输入张量。
• other：张量。用于取余的张量。
• out：张量。输出张量。

返回值

remainder() 方法返回一个张量，其中每个元素都被输入张量和用于取余的张量的对应元素取余。

示例

以下是使用 remainder() 方法的示例：

import torch

# define tensors
a = torch.tensor([5, 10, 15])
b = torch.tensor([2, 3, 4])

# calculate remainder
c = torch.remainder(a, b)

# print result
print(c)

执行上述代码会输出以下结果：

tensor([1, 1, 3])

本例中，a 和 b 都是张量，并且包含 3 个元素。按元素对它们执行取余操作，结果存储在 c 张量中。

总结

torch.remainder() 方法用于计算两个张量的元素之间的余数。输入张量和用于取余的张量必须具有相同的形状。