jax.Array.var — JAX documentation (original) (raw)

jax.Array.var#

abstract Array.var(axis=None, dtype=None, out=None, ddof=0, keepdims=False, *, where=None, correction=None)[source]#

Compute the variance along a given axis.

Refer to jax.numpy.var() for full documentation.

Parameters:

Return type:

Array