code: def broadcast_size(a, b):
      return torch.tensor(
          [max(i, j) for i, j in zip(a.shape, b.shape)]
      ).prod()
  
file_path: pytorch-struct/torch_struct/semirings/checkpoint.py
imports: import torch
label_desc: Use image node_id 1 for calling a global function with example usage: broadcast_size(a, b) and returns: torch