torch.func.replace\_all\_batch\_norm\_modules\_ =============================================== .. currentmodule:: torch.func .. autofunction:: replace_all_batch_norm_modules_