Method __init__

utils.py:73–76

(self, dim: int, eps: float = 1e-6)

    """

    def __init__(self, dim: int, eps: float = 1e-6) -> None:
        super().__init__()
        self.eps = eps
        self.dim = dim

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        # x: [..., dim]