torch_geopooling.functional

These are low-level primitives for geospatial learning.

Sparse functions

torch_geopooling.functional.embedding2d

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

Pooling functions

torch_geopooling.functional.avg_quad_pool2d

Average pooling over quadtree decomposition of input 2D coordinates.

torch_geopooling.functional.max_quad_pool2d

Maximum pooling over quadtree decomposition of input 2D coordinates.

torch_geopooling.functional.quad_pool2d

Lookup index over quadtree decomposition of input 2D coordinates.

Adaptive pooling functions

torch_geopooling.functional.adaptive_avg_quad_pool2d

Adaptive average pooling over quadtree decomposition of input 2D coordinates.

torch_geopooling.functional.adaptive_max_quad_pool2d

Adaptive maximum pooling over quadtree decomposition of input 2D coordinates.

torch_geopooling.functional.adaptive_quad_pool2d

Adaptive lookup index over quadtree decomposition of input 2D coordinates.