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


henry090/fastai documentation built on April 14, 2024, 9:28 a.m.