code: def forward(self, image1, image2):
      raise NotImplementedError
  
imports: from typing import Callable, NamedTuple, Optional
  import torch
  import torch.nn
  from torch import Tensor
  import neuralcompression.functional
label_desc: Use image node_id 2 for calling the DvcStage obj's underlying member method code with example usage: obj.forward(image1, image2) without return types