About Batchnorm

[ryujaehun]

Hi ,
TASO is only concerned about 2d batch norm case but some network has 1d or 3d batch norm case.

TASO/src/core/batchnorm.cc

Line 72 in f11782c

assert(_input.numDim == 4);

I have a question about the above code.
If I modify the above code, will it work rightly?