Method __init__

locationencoder/pe/direct.py:9–14

(self)

class Direct(nn.Module):
    def __init__(self):
        super(Direct, self).__init__()

        # adding this class variable is important to determine
        # the dimension of the follow-up neural network
        self.embedding_dim = 2

    def forward(self, coords):
        # place lon lat coordinates in a -pi, pi range