neural_tangents.stax.GlobalSumPool
- neural_tangents.stax.GlobalSumPool(batch_axis=0, channel_axis=- 1)[source]
Layer construction function for a global sum pooling layer.
Sums over and removes (keepdims=False) all spatial dimensions, preserving the order of batch and channel axes.
- Parameters
- Return type
Tuple
[InitFn
,ApplyFn
,LayerKernelFn
,MaskFn
]- Returns
(init_fn, apply_fn, kernel_fn).