def to_float(*args):
    """
    cast numpy arrays to float32
    if there's more than one, return an array
    """
    floats = [np.array(a, dtype=np.float32) for a in args]
    return floats[0] if len(floats) == 1 else floats