class_docstr: Independent multivariate Gaussian distribution parametrized by mean.
code: def _n_dims(self):
      return np.prod(self._shape, dtype=jnp.int32)
  
imports: import gin
  import gym
  import numpy
  from trax import layers
  from trax.fastmath import numpy
label_desc: Use image node_id 2 for calling the Gaussian obj's underlying member method code with example usage: obj._n_dims() and returns: np