diff options
author | soumith <soumith@fb.com> | 2015-08-05 15:26:45 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-08-05 15:27:33 +0300 |
commit | 9fe2e3a83c2f51107237a66ec035a303a8ba4141 (patch) | |
tree | 926ad8dd006d4d816a48ac17482043cbb089b4b1 /VolumetricMaxPooling.lua | |
parent | 7968df76ef9ec088b68ec5fe536f3804e8fb0ca3 (diff) |
Volumetric max and avg pooling
Diffstat (limited to 'VolumetricMaxPooling.lua')
-rw-r--r-- | VolumetricMaxPooling.lua | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/VolumetricMaxPooling.lua b/VolumetricMaxPooling.lua new file mode 100644 index 0000000..d200e36 --- /dev/null +++ b/VolumetricMaxPooling.lua @@ -0,0 +1,6 @@ +local VolumetricMaxPooling, parent = torch.class('cudnn.VolumetricMaxPooling', 'cudnn._Pooling3D') + +function VolumetricMaxPooling:__init(kT, kW, kH, dT, dW, dH, padT, padW, padH) + parent.__init(self, kT, kW, kH, dT, dW, dH, padT, padW, padH) + self.mode = 'CUDNN_POOLING_MAX' +end |