tf.keras.ops.stack

Join a sequence of tensors along a new axis.

Main aliases

tf.keras.ops.numpy.stack

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.