nip.utils.torch.GlobalMaxPool#
- class nip.utils.torch.GlobalMaxPool(dim: int = -1, keepdim: bool = False)[source]#
Global max pooling layer over a dimension.
- Parameters:
Methods Summary
__init__
([dim, keepdim])Initialize internal Module state, shared by both nn.Module and ScriptModule.
forward
(x)Apply global max pooling to the input tensor.
Attributes
T_destination
call_super_init
dump_patches
training
Methods