jax.numpy.stackΒΆ
-
jax.numpy.
stack
(arrays, axis=0, out=None)[source]ΒΆ Join a sequence of arrays along a new axis.
LAX-backend implementation of
stack()
.Original docstring below.
The
axis
parameter specifies the index of the new axis in the dimensions of the result. For example, ifaxis=0
it will be the first dimension and ifaxis=-1
it will be the last dimension.New in version 1.10.0.