nn$AvgPool layer for `ndim`

AvgPool(ks = 2, stride = NULL, padding = 0, ndim = 2, ceil_mode = FALSE)

Arguments

ks

kernel size

stride

the stride of the window. Default value is kernel_size

padding

implicit zero padding to be added on both sides

ndim

dimension number

ceil_mode

when True, will use ceil instead of floor to compute the output shape

Value

None