def METHOD_NAME(self, input_data):
    output_data = self._model.METHOD_NAME(input_data)
    if self.coae:
        real_output = self.coae.decode(output_data).detach().numpy()
        if real_output.shape[1] == 2:
            real_output = real_output[::, 1].reshape((-1, 1))
        return real_output
    else:
        return output_data