nn$AvgPool layer for `ndim`
AvgPool(ks = 2, stride = NULL, padding = 0, ndim = 2, ceil_mode = FALSE)
kernel size
the stride of the window. Default value is kernel_size
implicit zero padding to be added on both sides
dimension number
when True, will use ceil instead of floor to compute the output shape
None