code: def __init__(self, f, g):
      super().__init__()
      self.f = f
      self.g = g
  
imports: import torch
  import torch.nn
  from torch.autograd.function import Function
  from torch.utils.checkpoint import get_device_states, set_device_states
label_desc: Use image node_id 1 to create a new IrreversibleBlock object from inherited base classes: nn with example: obj = IrreversibleBlock(f, g)