JaxBackend.stack

JaxBackend.stack(*args, **kwargs)

Join a sequence of objects along a new axis.

Parameters:

arrBackendArray

Sequence of arrays.

axisint, optional

Axis along which to stack the input arrays.

Returns:

BackendArray

Stacked input data.