code: def __init__(self, model, output_op):
      super().__init__()
      self.model = model
      self.output_op = output_op
      assert output_op in [
          "argmax",
          "softmax",
      ], "output_op should in ['argmax', 'softmax']"
  
imports: import paddle
label_desc: Use image node_id 1 to create a new WrappedModel object from inherited base classes: paddle.nn with example: obj = WrappedModel(model, output_op)