Numpy: find the euclidean distance between two 3-D arrays,"np.linalg.norm(A - B, axis=-1)"