why return batch2d in inflate_batch_norm()?

[LiUzHiAn]

Hi,xiaolong,

Thanks for your appealing work. When you inflating the ResNet2d into ResNet3d, I noticed that you returned the batch2d object in >>inflate_batch_norm()<< function. Shouldn't it return the batch3d?