code: def test_normal_log_prob_shape():
      loc = torch.zeros(3, 2)
      scale = torch.ones(3, 2)
      x = torch.zeros(3, 2)
      d = dist.Normal(loc, scale)
      assert d.log_prob(x).size() == (3, 2)
  
file_path: pyro/tests/distributions/test_shapes.py
imports: import torch
  import pyro.distributions
label_desc: Use image node_id 9 for calling a global function with example usage: test_normal_log_prob_shape() without return types