tf_agents.networks.expand_dims_layer.ExpandDims

Expands dims along a particular axis.

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

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.