View source on GitHub |
Join a sequence of tensors along a new axis.
tf.keras.ops.stack(
x, axis=0
)
The axis
parameter specifies the index of the new axis in the
dimensions of the result.
Args | |
---|---|
x
|
A sequence of tensors. |
axis
|
Axis along which to stack. Defaults to 0 .
|
Returns | |
---|---|
The stacked tensor. |