Module: tf_agents.keras_layers.inner_reshape

Keras layer to reshape inner dimensions (keeping outer dimensions the same).

Functions

InnerReshape(...): Returns a Keras layer that reshapes the inner dimensions of tensors.

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