Часть 3. Далеко ли, близко ли?

import numpy as np

def get_mutual_l2_distances_vectorized(
    lhs: np.ndarray,
    rhs: np.ndarray,
) -> np.ndarray:
    res = np.sqrt(np.sum( (lhs[:, np.newaxis, :] - rhs[np.newaxis, :, :]) ** 2, axis = 2 ))
    return res