Add BatchNorm1d and BatchNorm2d ops support

Author: Esteban Gómez

docs/index.md

@@ -125,4 +125,6 @@ By default, all methods support all modules as long as these are instances of `t
 | `torch.nn.AdaptiveMaxPool2d` | :material-check:                   | 0.0.1   |
 | `torch.nn.MaxPool1d`         | :material-check:                   | 0.0.1   |
 | `torch.nn.MaxPool2d`         | :material-check:                   | 0.0.1   |
-| `torch.nn.LayerNorm`         | :material-check:                   | 0.0.1   |
+| `torch.nn.LayerNorm`         | :material-check:                   | 0.0.1   |
+| `torch.nn.BatchNorm1d`       | :material-check:                   | 0.0.4   |
+| `torch.nn.BatchNorm2d`       | :material-check:                   | 0.0.4   |

src/moduleprofiler/__init__.py

@@ -1,5 +1,5 @@
 "Module profiler"
-__version__ = "0.0.3"
+__version__ = "0.0.4"
 
 __all__ = [
     "get_default_ops_map",