jax_copy

probly.representation.jax_functions.jax_copy(a: JaxArrayLike, /, order: str | None = None) → jax.Array

Return a copy of the array, mirroring jax.numpy.copy.

Parameters:

• a – The array to copy.

• order – The memory layout of the copy.

Returns:

The copied array.