Other modules

MaxPoolChannels

class MaxPoolChannels(kernel_size)

Bases: torch.nn.modules.module.Module

Module that computes the maximum activation within each group of kernel_size consecutive channels.

Parameters

kernel_size (int) – the size of the group of channels the max is computed over