Method forward

locationencoder/pe/direct.py:16–19

(self, coords)

        self.embedding_dim = 2

    def forward(self, coords):
        # place lon lat coordinates in a -pi, pi range
        coords = torch.deg2rad(coords) - torch.pi
        return coords