StdMean

Calculates the standard deviation and mean.

Abstract Signature:

StdMean(input: Tensor, dim: Optional[Union[int, List[int]]], correction: int = 1, keepdim: bool = False)

PyTorchJAX (Core)

PyTorch

API: torch.std_mean

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

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