Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tf_agents.networks.expand_dims_layer

View source on GitHub

Keras layer performing the equivalent of tf.expand_dims.

Classes

class ExpandDims: Expands dims along a particular axis.