VarMean

Returns (var, mean).

Abstract Signature:

VarMean(x: Tensor, axis: int, correction: int = 1, keepdims: bool = False)

PyTorch

API: torch.var_mean
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: —
Strategy: Macro '(jax.numpy.var({x}, axis={axis}, ddof={correction}, keepdims={keepdims}), jax.numpy.mean({x}, axis={axis}, keepdims={keepdims}))'

NumPy

API: —
Strategy: Macro '(numpy.var({x}, axis={axis}, ddof={correction}, keepdims={keepdims}), numpy.mean({x}, axis={axis}, keepdims={keepdims}))'

Keras

API: —
Strategy: Macro '(keras.ops.var({x}, axis={axis}, keepdims={keepdims}), keras.ops.mean({x}, axis={axis}, keepdims={keepdims}))'

TensorFlow

API: tf.nn.moments
Strategy: Macro '(tf.math.reduce_variance({x}, axis={axis}, keepdims={keepdims}), tf.math.reduce_mean({x}, axis={axis}, keepdims={keepdims}))'

Apple MLX

API: —
Strategy: Macro '(mlx.core.var({x}, axis={axis}, ddof={correction}, keepdims={keepdims}), mlx.core.mean({x}, axis={axis}, keepdims={keepdims}))'

Flax NNX

API: —
Strategy: Macro '(jax.numpy.var({x}, axis={axis}, ddof={correction}, keepdims={keepdims}), jax.numpy.mean({x}, axis={axis}, keepdims={keepdims}))'

PaxML / Praxis

API: —
Strategy: Macro '(jax.numpy.var({x}, axis={axis}, ddof={correction}, keepdims={keepdims}), jax.numpy.mean({x}, axis={axis}, keepdims={keepdims}))'