广播机制 不同维度的张量可以通过广播机制进行运算,那什么是广播机制? 两个张量要可以进行运算,从右到左,要么维度相等,要么其中一个为1( 不存在的认作1)。 import torch # 从右到左,要么维度相等,要么其中一个为1,不存在的认作1 a=torch.randn(size=(2,1,1,4)) b=torch.randn(size=(4,2,4)) c=a+b print(c.size()) torch.Size([2, 4, 2, 4])