tf_agents.networks.expand_dims_layer.ExpandDims

Expands dims along a particular axis.

tf_agents.networks.expand_dims_layer.ExpandDims(
    axis, **kwargs
)

Args
axis	Axis to expand. A new dim is added before this axis. May be a negative value. Must not be a tensor.

Input shape
(batch_size,) + shape

Output shape
(batch_size,) + shape + [1], if axis == -1. (batch_size,) + shape[:axis + 1] + [1] + shape[axis + 1:], if axis < -1. (batch_size,) + shape[:axis] + [1] + shape[axis:], if axis >= 0.