# {py:mod}`evorl.distribution`

```{py:module} evorl.distribution
```

```{autodoc2-docstring} evorl.distribution
:parser: autodoc2_docstrings_parser
:allowtitles:
```

## Module Contents

### Classes

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`TanhTransformedDistribution <evorl.distribution.TanhTransformedDistribution>`
  - ```{autodoc2-docstring} evorl.distribution.TanhTransformedDistribution
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### Functions

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`get_categorical_dist <evorl.distribution.get_categorical_dist>`
  - ```{autodoc2-docstring} evorl.distribution.get_categorical_dist
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
* - {py:obj}`get_tanh_norm_dist <evorl.distribution.get_tanh_norm_dist>`
  - ```{autodoc2-docstring} evorl.distribution.get_tanh_norm_dist
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
* - {py:obj}`get_trancated_norm_dist <evorl.distribution.get_trancated_norm_dist>`
  - ```{autodoc2-docstring} evorl.distribution.get_trancated_norm_dist
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### Data

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`tfd <evorl.distribution.tfd>`
  - ```{autodoc2-docstring} evorl.distribution.tfd
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### API

`````{py:class} TanhTransformedDistribution(distribution, threshold=0.999, validate_args=False)
:canonical: evorl.distribution.TanhTransformedDistribution

Bases: {py:obj}`evorl.distribution.tfd`

```{autodoc2-docstring} evorl.distribution.TanhTransformedDistribution
:parser: autodoc2_docstrings_parser
```

````{py:method} entropy(seed=None)
:canonical: evorl.distribution.TanhTransformedDistribution.entropy

```{autodoc2-docstring} evorl.distribution.TanhTransformedDistribution.entropy
:parser: autodoc2_docstrings_parser
```

````

````{py:method} log_prob(event)
:canonical: evorl.distribution.TanhTransformedDistribution.log_prob

```{autodoc2-docstring} evorl.distribution.TanhTransformedDistribution.log_prob
:parser: autodoc2_docstrings_parser
```

````

````{py:method} mode()
:canonical: evorl.distribution.TanhTransformedDistribution.mode

```{autodoc2-docstring} evorl.distribution.TanhTransformedDistribution.mode
:parser: autodoc2_docstrings_parser
```

````

`````

````{py:function} get_categorical_dist(logits: jax.Array)
:canonical: evorl.distribution.get_categorical_dist

```{autodoc2-docstring} evorl.distribution.get_categorical_dist
:parser: autodoc2_docstrings_parser
```
````

````{py:function} get_tanh_norm_dist(loc: jax.Array, scale: jax.Array, min_scale: float = 0.001)
:canonical: evorl.distribution.get_tanh_norm_dist

```{autodoc2-docstring} evorl.distribution.get_tanh_norm_dist
:parser: autodoc2_docstrings_parser
```
````

````{py:function} get_trancated_norm_dist(loc, scale, low, high)
:canonical: evorl.distribution.get_trancated_norm_dist

```{autodoc2-docstring} evorl.distribution.get_trancated_norm_dist
:parser: autodoc2_docstrings_parser
```
````

````{py:data} tfd
:canonical: evorl.distribution.tfd
:value: >
   None

```{autodoc2-docstring} evorl.distribution.tfd
:parser: autodoc2_docstrings_parser
```

````