megengine.functional.normalize

normalize(inp, ord=None, axis=None, eps=1e-12)

返回在给定轴 axis 上 inp 张量每一行进行 \(L_p\) 归一化后的结果。

对于一个形如 \((n_0, ..., n_{dim}, ..., n_k)\) 的张量, 其每个 \(n_{dim}\) - axis 维度上的元素向量 \(v\) 将转化为:

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

参数

• inp (Tensor) – 输入张量。

• ord (Optional[float]) – 对 inp 进行 p 次幂计算. 默认：2

• axis (Optional[int]) – 要进行规约的维度。如果设置为 None，则所有的维度都将被规约。默认：None

• eps (float) – 为防止0做除数而设的较小值。 默认： 1e-12

返回类型

Tensor

返回

归一化的输出张量。

上一页

megengine.functional.norm

下一页

megengine.functional.gather