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 June 22, 2024, 11:15 a.m.