class_docstr: This class is a wrapper for numpy argsort function.
class_docstr_tok: ["This","class","is","a","wrapper","for","numpy","argsort","function","."]
code: def infer_shape(self, node, inputs_shapes):
      if _variable_is_none(node.inputs[1]):
          return [(mul(*inputs_shapes[0]),)]
      # axis should not be None, so there should be the same number of
      # dimensions in the input and output
      assert node.inputs[0].ndim == node.outputs[0].ndim
      assert inputs_shapes[1] == ()
      return [inputs_shapes[0]]
  
imports: from __future__ import absolute_import, print_function, division
  import numpy
  import theano
  from theano.tensor.basic import mul, arange
  from theano.gradient import grad_undefined
  from theano.tensor.subtensor import set_subtensor
label_desc: Use image node_id 5 for calling the ArgSortOp obj's underlying member method code with example usage: obj.infer_shape(node, inputs_shapes) and returns: list, list