def METHOD_NAME(self, x):
    """
    Used to get the token for the first
    :param x: An array with shape (N, input_size) where N is a whole number greater or equal to 1, and input_size is the input size used when creating the model.
    :return: An array with shape (N,) where N is the same as N from the input. Every number in the array is a whole number in range 0...output_size - 1 where output_size is the output size used when creating the model.
    """
    return torch.argmax(self(x), dim=1)