本文重点

在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