StdMean ======= Calculates the standard deviation and mean. **Abstract Signature:** ``StdMean(input: Tensor, dim: Optional[Union[int, List[int]]], correction: int = 1, keepdim: bool = False)`` .. raw:: html

PyTorch

API: torch.std_mean
Strategy: Direct Mapping

JAX (Core)

API:
Strategy: Macro '(jax.numpy.std({input}, axis={dim}, ddof={correction}, keepdims={keepdim}), jax.numpy.mean({input}, axis={dim}, keepdims={keepdim}))'