VolumetricAveragePooling.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

Welcome to mirror list, hosted at ThFree Co, Russian Federation.

VolumetricAveragePooling.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

summary refs log tree commit diff

path: root/VolumetricAveragePooling.lua

blob: bcebebac136d07224f0a1820e38ea0e30fd895ed (plain)

local VolumetricAveragePooling, parent = torch.class('cudnn.VolumetricAveragePooling', 'cudnn._Pooling3D')

function VolumetricAveragePooling:updateOutput(input)
   self.mode = 'CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING'
   return parent.updateOutput(self, input)
end