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.BatchNorm1d
、mindspore.nn.BatchNorm2d
、mindspore.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 ]]]]