Welcome to mirror list, hosted at ThFree Co, Russian Federation.
1 2 3 4
local BN, parent = torch.class('nn.VolumetricBatchNormalization', 'nn.BatchNormalization') -- expected dimension of input BN.nDim = 5