def get_distance(point_cloud):
    """
    :param point_cloud: (B, C, N) Point cloud data.
    :return: (B, N) Distance of each point in the point   cloud.
    """
    distance = point_cloud.pow(2).sum(axis=1).sqrt()
    return distance