diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-08 03:22:07 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-09 21:42:51 +0300 |
| commit | 4696a46212ffc5bb54b0d8f3327dfa47693bbfdf (patch) | |
| tree | 5f4512a8ead3d952d17a9a82c200a60ba6cb5082 /VolumetricMaxPooling.lua | |
| parent | 948ac6a26cc6c2812e04718911bca9a4b641020e (diff) | |
lazy init
Diffstat (limited to 'VolumetricMaxPooling.lua')
| -rw-r--r-- | VolumetricMaxPooling.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/VolumetricMaxPooling.lua b/VolumetricMaxPooling.lua index 8e2deaf..54eb39c 100644 --- a/VolumetricMaxPooling.lua +++ b/VolumetricMaxPooling.lua @@ -36,6 +36,7 @@ function VolumetricMaxPooling:floor() end function VolumetricMaxPooling:updateOutput(input) + self.indices = self.indices or input.new() input.THNN.VolumetricMaxPooling_updateOutput( input:cdata(), self.output:cdata(), |