cudnn.torch/VolumetricBatchNormalization.lua at master · soumith/cudnn.torch · GitHub

Sitelet https://github.com/soumith/cudnn.torch/blob/master/VolumetricBatchNormalization.lua

6 lines (5 loc) · 262 Bytes

local VolumetricBatchNormalization =
   torch.class('cudnn.VolumetricBatchNormalization', 'cudnn.BatchNormalization')
VolumetricBatchNormalization.mode = 'CUDNN_BATCHNORM_SPATIAL'
VolumetricBatchNormalization.nDim = 5
VolumetricBatchNormalization.__version = 2