Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tf_agents.networks.utils

View source on GitHub

Network utilities.

Classes

class BatchSquash: Facilitates flattening and unflattening batch dims of a tensor.

Functions

maybe_permanent_dropout(...): Adds a Keras dropout layer with the option of applying it at inference.

mlp_layers(...): Generates conv and fc layers to encode into a hidden state.