torch.nn.functional.normalize

torch.nn.functional.normalize(input, p=2.0, dim=1, eps=1e-12, out=None)

在指定維度上對輸入執行 $L_p$ 範數歸一化。

對於形狀為 $(n_0, ..., n_{dim}, ..., n_k)$ 的張量 input，沿著 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，此操作將不可微分。

返回型別

張量