code: def __init__(self, f: nn.Module, g: nn.Module, split_dim: int = -1):
      super().__init__()
      self.f = Deterministic(f)
      self.g = Deterministic(g)
      self.split_dim = split_dim
  
imports: from typing import List
  import torch
  import torch.nn
  from torch.autograd.function import Function
  from torch.utils.checkpoint import get_device_states, set_device_states
  from xformers.components import RequiresWrappedInputs
label_desc: Use image node_id 1 to create a new ReversibleBlock object from inherited base classes: nn with example: obj = ReversibleBlock(f, g, split_dim)