Qiuzhongyuan

torch::nn::BatchNorm -> torch::nn::BatchNorm2d ; conv_options.stride_ = stride -> conv_options.stride(stride) torch::nn::BatchNormImpl -> torch::nn::BatchNorm2dImpl and so on @xuweilin2014