code: def rmv(A, b):
      """Tensorized matrix vector multiplication of rightmost dimensions."""
      return torch.matmul(A, b.unsqueeze(-1)).squeeze(-1)
  
file_path: pyro/pyro/contrib/util.py
imports: import weakref
  from collections import OrderedDict
  import torch
  import pyro
label_desc: Use image node_id 4 for calling a global function with example usage: rmv(A, b) and returns: torch