code: def fast_collate(batch):
      targets = torch.tensor([b[1] for b in batch], dtype=torch.int64)
      batch_size = len(targets)
      tensor = torch.zeros(
          (batch_size, *batch[0][0].shape), dtype=torch.uint8
      )
      for i in range(batch_size):
          tensor[i] += torch.from_numpy(batch[i][0])
  
      return tensor, targets
  
file_path: gen-efficientnet-pytorch/data/loader.py
imports: import torch
  import torch.utils.data
  from .transforms import *
label_desc: Use image node_id 1 for calling a global function with example usage: fast_collate(batch) and returns: tensor,  targets