ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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.