View source on GitHub
|
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.