torch.nn.functional.normalize

torch.nn.functional.normalize(input, p=2.0, dim=1, eps=1e-12, out=None) [исходный код]

Выполняет нормализацию $L_p$ входных данных по указанному измерению.

Для тензора input размера $(n_0, ..., n_{dim}, ..., n_k)$, каждый вектор из $n_{dim}$ элементов $v$ вдоль измерения dim преобразуется следующим образом

$$v = \frac{v}{\max(\lVert v \rVert_p, \epsilon)}.$$

При аргументах по умолчанию для нормализации используется евклидова норма векторов вдоль измерения $1$.

Параметры:

• input (Tensor) – входной тензор произвольной формы
• p (float) – значение показателя степени в формуле нормы. По умолчанию: 2
• dim (int or tuple of ints) – измерение, по которому выполняется свёртка. По умолчанию: 1
• eps (float) – малое значение для предотвращения деления на ноль. По умолчанию: 1e-12
• out (Tensor, optional) – выходной тензор. Если используется out, эта операция не будет дифференцируемой.

Тип возвращаемого значения:

Tensor