def scale_pc(data, scale):
    """ scale the point cloud
    Input:
        data: (B,num_points,3) tensor
        scale: scalar tensor
    Return:
        (B,num_points,3) tensor, scaled point cloud
    """
    return data * scale