megengine.functional.normalize

normalize(inp, ord=None, axis=None, eps=1e-12)[源代码]

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

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

\[v = \frac{v}{\max(\lVert v \rVert_p, \epsilon)}.\]
参数
  • inp – 输入张量。

  • ord – 对 inp 进行 p 次幂计算. 默认:2

  • axis – 要进行规约的维度。如果设置为 None,则所有的维度都将被规约。默认:None

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

返回

归一化的输出张量。