mindspore.nn.LRN

查看源文件
class mindspore.nn.LRN(depth_radius=5, bias=1.0, alpha=1.0, beta=0.5, norm_region='ACROSS_CHANNELS')[源代码]

局部响应归一化操作LRN(Local Response Normalization)。

警告

LRN在Ascend平台已废弃,存在潜在精度问题。建议使用其他归一化方法,如 mindspore.nn.BatchNorm1dmindspore.nn.BatchNorm2dmindspore.nn.BatchNorm3d 代替LRN。

更多参考详见 mindspore.ops.lrn()

支持平台:

GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, nn
>>> import numpy as np
>>> input_x = Tensor(np.array([[[[0.1], [0.2]],
...                       [[0.3], [0.4]]]]), mindspore.float32)
>>> output = nn.LRN()(input_x)
>>> print(output)
[[[[0.09534626]
   [0.1825742 ]]
  [[0.2860388 ]
   [0.3651484 ]]]]