PoolFlatten: PoolFlatten

View source: R/layers.R

PoolFlattenR Documentation

PoolFlatten

Description

Combine 'nn.AdaptiveAvgPool2d' and 'Flatten'.

Usage

PoolFlatten(pool_type = "Avg")

Arguments

pool_type

pooling type

Value

None


fastai documentation built on March 21, 2022, 9:07 a.m.