code: def rmm(A, B):
      """Shorthand for `matmul`."""
      return torch.matmul(A, B)
  
file_path: pyro/pyro/contrib/util.py
imports: import weakref
  from collections import OrderedDict
  import torch
  import pyro
label_desc: Use image node_id 3 for calling a global function with example usage: rmm(A, B) and returns: torch