Method reset_parameters

python/oneflow/nn/modules/normalization.py:147–150

(self)

        self.reset_parameters()

    def reset_parameters(self) -> None:
        if self.affine:
            flow.nn.init.ones_(self.weight)
            flow.nn.init.zeros_(self.bias)

    def forward(self, input: Tensor) -> Tensor:
        return group_norm(