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