code: def __init__(self, distance, iter_fn=None, batch_size=32):
      super().__init__()
      self.distance = distance
      self.iter_fn = iter_fn
      self.batch_size = batch_size
  
imports: import torch
label_desc: Use image node_id 1 to create a new BatchedDistance object from inherited base classes: torch.nn with example: obj = BatchedDistance(distance, iter_fn, batch_size)