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 31, 2023, 11:41 p.m.