####### Modules ####### .. toctree:: :maxdepth: 1 batch_norm conv conv_transpose embedding group_norm layer_norm linear