code: def __init__(
      self,
      num_features: int,
      *,
      eps: float = 1e-5,
      momentum: float = 0.1,
  ) -> None:
      super().__init__()
      self.bn = TS.nn.BatchNorm(
          num_features=num_features, eps=eps, momentum=momentum
      )
  
file_path: torch-points3d/torch_points3d/modules/SparseConv3d/nn/torchsparse.py
imports: import torch
  import torchsparse
  import torchsparse.nn
label_desc: Use image node_id 1 to create a new BatchNorm object from inherited base classes: torch.nn with example: obj = BatchNorm(num_features)