Method forward

astroclip/modules.py:290–293

(self, input)

            torch.nn.init.zeros_(self.bias)

    def forward(self, input):
        return F.layer_norm(
            input, self.normalized_shape, self.weight, self.bias, self.eps
        )


class TiedLinear(nn.Module):