#广播机制的运用 #需求:将一个形状为13x13x3x2的特征图每一个点位表示的2维向量,与10x2 十个二维向量按位相乘后求和。 #广播机制的实质:广播机制是内容复制和按位运算,先把两个tensor的形状复制到一样,然后按位运算。 #比如a=4x1x3和b=5x3之间做位运算,首先对比右边的形状,相同。然后对比倒数第二个,把A的1x3复制5份,变成4x5x3.再把B的5x3复制4份,变成4x5x3。这样就能够按位运算了。 a=np.random.normal(size=(13,13,3,2)) b=np.random.normal(size=(10,2)) a=np.expand_dims(a,axis=-2)#形状不满足的,通过扩展维度,变成(13,13,3,1,2) ab=a+b print("")
广播机制是怎么回事?
最新推荐文章于 2023-08-28 17:05:16 发布