Layer Normalization 总览 针对同一通道数的图片的H*W进行层正则化,后面的γ和β是可以学习的参数,其中这两个的维度和最后一个的维度相同 例如特征图矩阵维度为[3, 577, 768], 那么γ和β的维度均为Tensor(768,) step1:代码示例1 import torch import torch.nn as nn input = torch.tensor( [ [ [ [2., 2.], [3., 3.] ], [ [3., 3.], [2., 2.] ] ], […