Save the date! Google I/O returns May 18-20 Register now

Module: tf_agents.networks.expand_dims_layer

Keras layer performing the equivalent of tf.expand_dims.

Classes

class ExpandDims: Expands dims along a particular axis.