tf.keras.ops.stack

Join a sequence of tensors along a new axis.

The axis parameter specifies the index of the new axis in the dimensions of the result.

x A sequence of tensors.
axis Axis along which to stack. Defaults to 0.

The stacked tensor.