def METHOD_NAME(self, x):
    outs = self.net(x)
    new_outs = []
    for out in outs:
        if self.output_op == 'argmax':
            out = paddle.argmax(out, axis=1, dtype='int32')
        elif self.output_op == 'softmax':
            out = paddle.nn.functional.softmax(out, axis=1)
        new_outs.append(out)
    return new_outs