class_docstr: This class is a wrapper for numpy sort function.
class_docstr_tok: ["This","class","is","a","wrapper","for","numpy","sort","function","."]
code: def __get_expanded_dim(self, a, axis, i):
      index_shape = [1] * a.ndim
      index_shape[i] = a.shape[i]
      # it's a way to emulate
      # numpy.ogrid[0: a.shape[0], 0: a.shape[1], 0: a.shape[2]]
      index_val = arange(a.shape[i]).reshape(index_shape)
      return index_val
  
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 7 for calling the SortOp obj's underlying member method code with example usage: obj.__get_expanded_dim(a, axis, i) and returns: index_val