每天五分钟深度学习pytorch：pytorch中的广播机制是什么？

作者：羊村懒王 | 2024-06-13 01:36:17

踩

本文重点

在pytorch中经常有张量和张量之间的运算，那么有一点需要注意，那就是维度要匹配，如果维度不匹配就有可能出现问题。如果维度不一致，此时也可以同时进行操作，此时就需要使用pytorch中的广播机制，本节课程就讲解pytorch中的广播机制。

如上就是pytorch的广播机制，当两个张量之间维度不一致的时候，此时广播机制会起作用

广播机制的规则对dim小的向量来进行扩展，比如说，一个向量的维度是（16，3，28，28）

而另外一个向量的维度是（1），那么它会对（1）这个向量来进行扩展，步骤是这样的：

首先先将向量（1）向高维度扩展，扩展成（1，1，1，1），注意（1，1，1，1）这四个1中越往后表示维度越低，也就是说我们要将向量（1）的前面加上三个1

变为（1，1，1，1）之后，然后再扩展成（16，3，28，28），然后就可以进行对位相加了

A：（4&#x

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/710404