stack

Signature

nabla.stack(arrays: list[nabla.core.array.Array], axis: int = 0) -> nabla.core.array.Array

Description

Stack arrays along a new axis.

Parameters

arrays: List of arrays to stack axis: Axis along which to stack the arrays (default: 0)

Returns

Stacked array