![]() |
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
.