Module: tf_agents.networks.expand_dims_layer

Stay organized with collections Save and categorize content based on your preferences.

Keras layer performing the equivalent of tf.expand_dims.

Classes

class ExpandDims: Expands dims along a particular axis.

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