freezeBatchNorm

fun freezeBatchNorm(scaleShift: DTensor, mean: DTensor, variance: DTensor): AffineTransform