from bn import batch_norm
print(batch_norm)