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.