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)$，沿著維度 dim 的每個 $n_{dim}$ 元素向量 $v$ 的轉換方式如下：

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

使用預設參數時，它使用沿著維度 $1$ 的歐幾里得範數來進行正規化。

參數

• input (Tensor) – 任何形狀的輸入張量

• p (float) – 範數公式中的指數值。預設值：2

• dim (int 或 tuple of ints) – 要縮減的維度。預設值：1

• eps (float) – 避免除以零的小值。預設值：1e-12

• out (Tensor, optional) – 輸出張量。如果使用 out，則此操作將不可微分。

返回類型

Tensor