BatchNorm2D
case class BatchNorm2D(scope: Scope, input: Variable, weight: Variable, bias: Variable, runningMean: STen, runningVar: STen, training: Boolean, momentum: Double, eps: Double) extends Op
Batch Norm 2D 0-th dimension are samples. 1-th are features, everything else is averaged out.