class_docstr: Base class for pytorch module with data-aware initializer on first batch.
class_docstr_tok: ["Base","class","for","pytorch","module","with","data-aware","initializer","on","first","batch","."]
code: def __init__(self):
      super().__init__()
      self._is_initialized_tensor = nn.Parameter(
          torch.tensor(0, dtype=torch.uint8), requires_grad=False
      )
      self._is_initialized_bool = None
  
file_path: pytorch_tabular/src/pytorch_tabular/models/common/layers/misc.py
imports: from typing import Callable, Union
  import torch
  from torch import nn
label_desc: Use image node_id 1 to create a new ModuleWithInit object from inherited base classes: nn with example: obj = ModuleWithInit()