硫辨姥 发表于 2025-9-24 17:50:09

PyTorch中的求余

技术背景

求余数,是一个比较简单的数学计算,内容就不展开介绍了。在不同的工具中,在实现余数计算这个操作的时候,有可能有不同的结果。这里介绍的是Python、Numpy和PyTorch中的几种不同的计算余数的方法。
计算示例

首先最简单的,演示一下Python原生的余数计算方法:
In : -0.5%5
Out: 4.5这个意思就是说,-0.5对5求余数,得到的结果是4.5,那么也是数学上所预期的正确结果。然后看一下numpy的余数计算:
In : import numpy as np

In : a = np.array([-0.5])

In : b = np.array()

In : a%b
Out: array()

In : np.mod(a,b)
Out: array()numpy的计算结果跟数学上的预期是一致的,不论是哪个函数。接下来再看一下pytorch的几个求余数接口:
In : from torch import tensor, fmod, remainder

In : a = tensor([-0.5])

In : b = tensor()

In : a%b
Out: tensor()

In : fmod(a,b)
Out: tensor([-0.5000])

In : remainder(a,b)
Out: tensor()其中比较特殊的一个是fmod函数。从数学上来说,余数我们一般只取正数,但是fmod的计算结果可以有负数:
In : from torch import tensor, fmod

In : a = tensor([-5.5])

In : b = tensor()

In : fmod(a,b)
Out: tensor([-0.5000])

In : a = tensor([-4.9])

In : fmod(a,b)
Out: tensor([-4.9000])因此,pytorch中的remainder函数才是我们正常所理解的求余数操作,也直接使用a%b这样的写法。
总结概要

本文通过几个示例,介绍了在Python、Numpy和PyTorch三个不同的框架下,对于求余数函数的定义。比较特殊的是pytorch中的fmod函数,并不符合数学上的求余数方法,而是需要使用remainder函数。
版权声明

本文首发链接为:https://www.cnblogs.com/dechinphy/p/torch_mod.html
作者ID:DechinPhy
更多原著文章:https://www.cnblogs.com/dechinphy/
请博主喝咖啡:https://www.cnblogs.com/dechinphy/gallery/image/379634.html

来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

辜酗徇 发表于 2025-12-16 04:31:51

东西不错很实用谢谢分享

焦听云 发表于 2025-12-18 08:39:28

谢谢分享,辛苦了

嗣伐 发表于 2025-12-28 04:33:19

感谢分享,下载保存了,貌似很强大

诘琅 发表于 2026-1-14 21:31:05

感谢,下载保存了

搁胱 发表于 2026-1-15 05:35:51

分享、互助 让互联网精神温暖你我

敛饺乖 发表于 2026-1-15 23:49:34

不错,里面软件多更新就更好了

欧阳梓蓓 发表于 2026-1-18 21:28:36

懂技术并乐意极积无私分享的人越来越少。珍惜

锟及 发表于 2026-1-19 09:45:36

感谢发布原创作品,程序园因你更精彩

骂治并 发表于 2026-1-20 07:39:15

这个好,看起来很实用

遇玷 发表于 2026-1-20 19:40:24

懂技术并乐意极积无私分享的人越来越少。珍惜

钱艷芳 发表于 2026-1-21 10:20:30

感谢分享,学习下。

柴古香 发表于 2026-1-21 16:52:33

这个有用。

梨恐 发表于 2026-1-22 22:31:10

感谢分享,下载保存了,貌似很强大

僚娥 发表于 2026-1-24 10:28:49

喜欢鼓捣这些软件,现在用得少,谢谢分享!

砂歹汤 发表于 2026-1-26 18:19:21

这个有用。

痨砖 发表于 2026-1-28 05:14:21

收藏一下   不知道什么时候能用到

押疙 发表于 2026-1-28 09:06:01

不错,里面软件多更新就更好了

翁真如 发表于 2026-1-29 01:55:28

感谢分享

幌斛者 发表于 2026-1-31 23:29:55

很好很强大我过来先占个楼 待编辑
页: [1] 2 3
查看完整版本: PyTorch中的求余