Renorm

Renormalizes sub-tensors along a dimension.

Abstract Signature:

Renorm(input: Tensor, p: float, dim: int, maxnorm: float)

PyTorchJAX (Core)

PyTorch

API: torch.renorm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (renorm_impl)