记录深度学习（三）----张量的广播和科学计算

1、张量的广播

张量的广播与numpy中的广播类似，存在以下两种情况：

• 两个张量至少有一个维度值相同，不同的维度值之中有一个必为1才能广播
• 两个张量所有的维度都不同时，两个张量其它维度必为1才能广播

a  = torch.zeros(12).reshape(3,4)
b = torch.ones(3).reshape(3,1)
c = torch.ones(3).reshape(1,3)
d = torch.ones(8).reshape(2,4)

a + b 

将 张量b的列复制4次然后再与a中的每个对应元素相加。

a + c 

b + c

同时将张量b 的列和c 的行复制3次都变成3行3列的2维张量再进行计算。

a +d