tf.keras.ops.expand_dims

Expand the shape of a tensor.

Insert a new axis at the axis position in the expanded tensor shape.

x Input tensor.
axis Position in the expanded axes where the new axis (or axes) is placed.

Output tensor with the number of dimensions increased.