nn()$AdaptiveAvgPool layer for `ndim`

AdaptiveAvgPool(sz = 1, ndim = 2)

Arguments

sz

size

ndim

dimension size