poolForward

wrapper of cudnnPoolingForward for Variable

version(grain_cuda)

poolForward

(

bool isMax = true

bool isAveragePad = false

T

size_t _tensorDims

size_t _poolDims

)

(

Variable!(T, _tensorDims, DeviceStorage) input

,

int[_poolDims] windowDim

,

int[_poolDims] padding

,

int[_poolDims] stride

,

T alpha = 1

,

T beta = 0

)

Meta

Source

See Implementation