code: def matmul(a, b, dims=1):
      a2, b2, size = broadcast(a, b)
      return genbmm.maxbmm(a2, b2).view(size)
  
file_path: pytorch-struct/torch_struct/semirings/fast_semirings.py
imports: import torch
  import torch.distributions
  from .semirings import _BaseLog
  from .sample import _SampledLogSumExp
label_desc: Use image node_id 2 for calling the FastMaxSemiring obj's underlying member method code with example usage: obj.matmul(a, b, dims) and returns: genbmm