code: def forward(self, x):
      outs = self.model(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
  
imports: import paddle
label_desc: Use image node_id 2 for calling the WrappedModel obj's underlying member method code with example usage: obj.forward(x) and returns: new_outs