Pytorch中的广播机制

最新推荐文章于 2025-04-26 12:39:35 发布

打呼噜的星.

最新推荐文章于 2025-04-26 12:39:35 发布

阅读量268

点赞数 2

分类专栏：深度学习&Pytorch 文章标签： pytorch python

本文链接：https://blog.youkuaiyun.com/wsj_jerry521/article/details/120590873

版权

深度学习&Pytorch 专栏收录该内容

10 篇文章

订阅专栏

本文深入探讨了Python中数组操作的广播机制，当两个数组的shape不匹配时，如何通过广播使它们能够进行数学运算。条件包括：每个张量至少有一个维度，以及特定的维度匹配规则。例如，如果A.shape=(15,10,4)，B.shape=(1,1,4)，那么A+B的shape将是(15,10,4)。了解广播对于高效使用NumPy和其他科学计算库至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

广播机制触发：当两个数组的shape不一样时候，1)程序会自动执行操作使得A.shape==B.shape, 2)对应位置进行相加
运算结果的shape是:A.shape和B.shape对应位置的最大值,比如:A.shape=(4,10,4),B.shape=(15,1,4),那么A+B的shape是(15,10,4)
并非任意两个数组能进行广播。
两种情况能进行广播:
（1）每个张量至少有一个维度
（2.1）A.ndim > B.ndim, 并且A.shape最后几个元素包含B.shape,
如：
A.shape=(2,3,4,5), B.shape=(3,4,5)
A.shape=(2,3,4,5), B.shape=(4,5)
A.shape=(2,3,4,5), B.shape=(5)
（2.2）当A.shape=B.shape, 并且A,B的每个shape元素对齐从后往前数，相同或其中某个为1或其中一个不存在
A.shape=(1,9,4), B.shape=(15,1,4)
A.shape=(1,9,4), B.shape=(15,1,1)
A.shape(5,3,4,1), B.shape=( ,3,1,1)