torch_geopooling.nn

These are basic building blocks for graphs.

Sparse Layers

torch_geopooling.nn.Embedding2d

Retrieves spatial embeddings from a fixed-size lookup table based on 2D coordinates.

Pooling Layers

torch_geopooling.nn.QuadPool2d	Lookup index over quadtree decomposition of input 2D coordinates.
torch_geopooling.nn.AvgQuadPool2d	Average pooling over quadtree decomposition of input 2D coordinates.
torch_geopooling.nn.MaxQuadPool2d	Maximum pooling over quadtree decomposition of input 2D coordinates.

Adaptive Pooling Layers

torch_geopooling.nn.AdaptiveQuadPool2d	Adaptive lookup index over quadtree decomposition of input 2D coordinates.
torch_geopooling.nn.AdaptiveMaxQuadPool2d	Adaptive maximum pooling over quadtree decomposition of input 2D coordinates.
torch_geopooling.nn.AdaptiveAvgQuadPool2d	Adaptive average pooling over quadtree decomposition of input 2D coordinates.