class_docstr: Gradient wrt x of the Softmax Op.
class_docstr_tok: ["Gradient","wrt","x","of","the","Softmax","Op","."]
code: def infer_shape(self, fgraph, node, shape):
      return [shape[1]]
  
imports: from textwrap import dedent
  import numpy
  import scipy
  from pytensor.graph.basic import Apply
  from pytensor.graph.replace import _vectorize_node
  from pytensor.link.c.op import COp
  from pytensor.tensor.basic import as_tensor_variable
  from pytensor.tensor.elemwise import get_normalized_batch_axes
  from pytensor.tensor.math import gamma, gammaln, neg, sum
label_desc: Use image node_id 5 for calling the SoftmaxGrad obj's underlying member method code with example usage: obj.infer_shape(fgraph, node, shape) and returns: list