Method __init__

s3f/gvp_layer.py:207–210

(self, dims)

    Takes tuples (s, V) as input and as output.
    '''
    def __init__(self, dims):
        super().__init__()
        self.s, self.v = dims
        self.scalar_norm = nn.LayerNorm(self.s)
        
    def forward(self, x):
        '''