def __init__(self, num_labels, *args, **kwargs):
    """Initializes a relative location task.

    Args:
      num_labels: integer, number of orientations to bin the relative