torch.nn.functional.normalize — PyTorch 1.10.1 documentation
pytorch.org › torchtorch.nn.functional.normalize torch.nn.functional.normalize(input, p=2.0, dim=1, eps=1e-12, out=None) [source] Performs L_p Lp normalization of inputs over specified dimension. For a tensor input of sizes (n_0, ..., n_ {dim}, ..., n_k) (n0 ,...,ndim ,...,nk ), each n_ {dim} ndim -element vector v v along dimension dim is transformed as
torch.norm — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.norm(input, p='fro', dim=None, keepdim=False, out=None, dtype=None) [source] Returns the matrix norm or vector norm of a given tensor. Warning torch.norm is deprecated and may be removed in a future PyTorch release. Its documentation and behavior may be incorrect, and it is no longer actively maintained.