Cannot Find Reference Batch Normalization In Init Python Syntax