广播机制broadcast
两个变量格式不统一,有时也可以运算:
情况1:两个数组各维度大小从后往前比对均一致。
如:
A = np.zeros((2,5,3,4))
B = np.zeros((3,4))
print((A+B).shape) # 输出 (2, 5, 3, 4)
A = np.zeros((4))
B = np.zeros((3,4))
print((A+B).shape) # 输出(3,4)
情况2:两个数组存在一些维度大小不相等时,有一个数组的该不相等维度大小为1
A = np.zeros((2.
转载
2022-04-08 14:22:16 ·
111 阅读 ·
0 评论