'For compatibility with functions used as model definitions (taking
an input tensor and returning the tensor giving the output
of the model on that input).'

def __call__(self, *args, **kwargs):

    return self.get_probs(*args, **kwargs)
