代码如下:
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 循环很消耗时间