Module: tf_agents.networks.utils

Network utilities.

Classes

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

Functions

check_single_floating_network_output(...)

clone_initializer(...)

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

Other Members
absolute_import	Instance of __future__._Feature
division	Instance of __future__._Feature
print_function	Instance of __future__._Feature