# Module: tf_agents.distributions.masked

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/distributions/masked.py) |

Define distributions for spaces where not all actions are valid.

## Classes

[`class MaskedCategorical`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/distributions/masked/MaskedCategorical): A categorical distribution which supports masks per step.