Functional pooling¶

torch_lattice.nn.functional.pooling.avg_pool3d(inputs, **kwargs)[source]¶

torch_lattice.nn.functional.pooling.global_avg_pool(inputs)[source]¶

torch_lattice.nn.functional.pooling.global_max_pool(inputs)[source]¶

torch_lattice.nn.functional.pooling.max_pool3d(inputs, **kwargs)[source]¶

torch_lattice.nn.functional.pooling.pool3d(inputs, *, mode, kernel_size=2, stride=2, padding=0, dilation=1)[source]¶

Local sparse 3D pooling over convolution-style neighborhoods.

Return type:

SparseTensor

Parameters:

torch_lattice.nn.functional.pooling.sum_pool3d(inputs, **kwargs)[source]¶