6.tensor的自动广播机制与向量化

代码如下：

import torch

if __name__ == '__main__':

    # 1.广播机制是指在不同形状的tensor进行计算时，可自动扩展到较大的相同形状进行计算
      #广播机制的要求：（1）两个矩阵的维度都不能为0
                   #（2） 若两个矩阵对应的维度不等时，不等的其中一个维度一定等于1
                   #（3)一个维度存在，另一个维度不存在时，也可以用广播机制
    a=torch.ones(3,1,4,1,2)
    b=torch.ones(4,1,2,1)
    c=a+b#[3, 4, 4, 2, 2]
    print(c.shape)

    #2.向量化操作
    #在计算时尽量用向量进行计算操作，用for 循环很消耗时间