R/layers.R
PoolFlatten.Rd
Combine `nn.AdaptiveAvgPool2d` and `Flatten`.
PoolFlatten(pool_type = "Avg")
pooling type
None