VarMean ======= Returns (var, mean). **Abstract Signature:** ``VarMean(x: Tensor, axis: int, correction: int = 1, keepdims: bool = False)`` .. raw:: html

PyTorch

API: torch.var_mean
Strategy: Direct Mapping

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}))'