code: def xy_model():
      d = dist.Bernoulli(0.5)
      x_axis = pyro.plate("x_axis", 2, dim=-1)
      y_axis = pyro.plate("y_axis", 3, dim=-2)
      pyro.sample("b", d)
      with x_axis:
          pyro.sample("bx", d.expand_by([2]))
      with y_axis:
          pyro.sample("by", d.expand_by([3, 1]))
      with x_axis, y_axis:
          pyro.sample("bxy", d.expand_by([3, 2]))
  
file_path: pyro/tests/infer/test_util.py
imports: import math
  import pytest
  import torch
  import pyro
  import pyro.distributions
  import pyro.poutine
  from pyro.infer.importance import psis_diagnostic
  from pyro.infer.util import MultiFrameTensor
  from tests.common import assert_equal
label_desc: Use image node_id 1 for calling a global function with example usage: xy_model() without return types