# Module: tf_agents.keras_layers.squashed_outer_wrapper

<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/keras_layers/squashed_outer_wrapper.py) |

SquashedOuterWrapper Keras Layer.

## Classes

[`class SquashedOuterWrapper`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/keras_layers/SquashedOuterWrapper): Squash the outer dimensions of input tensors; unsquash outputs.