minuet.nn.pooling#

Classes

GlobalAvgPool(*args, **kwargs)

Applies global_avg_pool() on the given SparseTensor x.

GlobalMaxPool(*args, **kwargs)

Applies global_max_pool() on the given SparseTensor x.

GlobalSumPool(*args, **kwargs)

Applies global_sum_pool() on the given SparseTensor x.

class GlobalAvgPool(*args, **kwargs)#

Applies global_avg_pool() on the given SparseTensor x.

class GlobalMaxPool(*args, **kwargs)#

Applies global_max_pool() on the given SparseTensor x.

class GlobalSumPool(*args, **kwargs)#

Applies global_sum_pool() on the given SparseTensor x.