def __init__(self, begin_norm_axis, begin_params_axis):
    super(LayerNormNet, self).__init__()
    self.norm = P.LayerNorm(begin_norm_axis, begin_params_axis)
def __init__(self, begin_norm_axis, begin_params_axis):
    super(LayerNormNet, self).__init__()
    self.norm = P.LayerNorm(begin_norm_axis, begin_params_axis)