def vecs_from_tensor(x):  # shape (...)
    """Converts from tensor of shape (3,) to Vecs."""
    # num_components = x.shape[-1]
    # assert num_components == 3
    return x[..., 0], x[..., 1], x[..., 2]