class_docstr: Module that produces a TanhTransformedDistribution distribution.
class_docstr_tok: ["Module","that","produces","a","TanhTransformedDistribution","distribution","."]
code: def __call__(self, inputs: jnp.ndarray) -> tfd.Distribution:
      loc = self._loc_layer(inputs)
      scale = self._scale_layer(inputs)
      scale = jax.nn.softplus(scale) + self._min_scale
      distribution = tfd.Normal(loc=loc, scale=scale)
      return tfd.Independent(
          TanhTransformedDistribution(distribution),
          reinterpreted_batch_ndims=1,
      )
  
imports: from typing import Any, List, Optional, Sequence, Union, Callable
  import chex
  import haiku
  import jax
  import jax.numpy
  import numpy
  import tensorflow_probability
  import tensorflow_probability.substrates.jax
label_desc: Use image node_id 2 for calling the NormalTanhDistribution obj's underlying member method code with example usage: obj.__call__(inputs) and returns: tfd