jax.numpy.array_split¶
-
jax.numpy.
array_split
(ary, indices_or_sections, axis=0)[source]¶ Split an array into multiple sub-arrays.
LAX-backend implementation of
array_split()
. Original docstring below.Pl]
- Parameters
axis (
int
) –
jax.numpy.
array_split
(ary, indices_or_sections, axis=0)[source]¶Split an array into multiple sub-arrays.
LAX-backend implementation of array_split()
.
Original docstring below.
Pl]
axis (int
) –