class_docstr: OneHotCategorical Distribution.
  
  Args:
      p(:class:`~chainer.Variable` or :ref:`ndarray`): Parameter of
          distribution.
code: def batch_shape(self):
      return self.p.shape[:-1]
  
imports: import chainer
  from chainer.backends import cuda
  from chainer import distribution
  from chainer.functions.math import exponential
  import chainer.functions.math.sum
  from chainer.utils import cache
label_desc: Use image node_id 4 for calling the OneHotCategorical obj's underlying member method code with example usage: obj.batch_shape() and returns: self