jax_std

probly.representation.jax_functions.jax_std(a: JaxArrayLike, /, axis: int | Sequence[int] | None = None, dtype: DTypeLike | None = None, out: None = None, ddof: int = 0, keepdims: bool = False, *, where: JaxArrayLike | None = None) → jax.Array[source]

Compute the standard deviation, mirroring jax.numpy.std.

Parameters:

  • a – The array to reduce.

  • axis – The axis or axes to reduce.

  • dtype – The data type of the accumulator.

  • out – Unsupported by JAX, must be None.

  • ddof – The delta degrees of freedom.

  • keepdims – Whether reduced axes are retained with size one.

  • where – Optional mask selecting the elements to include.

Returns:

The standard deviation of the array.