VolumetricBatchNormalization.lua « nn « torch « contrib - rspamd.git - Rapid spam filtering system: https://github.com/rspamd/rspamd

index : rspamd.git

Rapid spam filtering system: https://github.com/rspamd/rspamd

www-data

about summary refs log tree commit diff stats

path: root/contrib/torch/nn/VolumetricBatchNormalization.lua

blob: 6168a9245dcefa6b13051368ddcceacbbb11161c (plain)

local BN, parent = torch.class('nn.VolumetricBatchNormalization', 'nn.BatchNormalization')

-- expected dimension of input
BN.nDim = 5