Pytorch深度学习——广播机制与in-place操作

目录

In-place

广播机制

​ 代码

---

In-place

即直接将结果赋值给参与运算的某个变量

广播机制

主要用于将两个维度不同的tensor进行运算，需要满足右对齐，即从右往左看，两个tensor上下两个相对应的维度的数值进行对比，要么相等，要么需要满足其中一个维度为1

广播运算可以在搭建网络结构中简化运算

 

代码

import torch

a = torch.rand(1, 2, 1, 2, 3)
b = torch.rand(4, 2, 3)  # 这个b的维度后两维度与a相同，从右数第三个维度a的为1，则可以广播
c = a + b
print(a)
print(b)
print(c)
print(a.shape)
print(b.shape)
print(c.shape)