code: def test_log_I1_shapes():
      assert_equal(log_I1(10, tensor(0.6)).shape, torch.Size([11, 1]))
      assert_equal(log_I1(10, tensor([0.6])).shape, torch.Size([11, 1]))
      assert_equal(
          log_I1(10, tensor([[0.6]])).shape, torch.Size([11, 1, 1])
      )
      assert_equal(
          log_I1(10, tensor([0.6, 0.2])).shape, torch.Size([11, 2])
      )
      assert_equal(log_I1(0, tensor(0.6)).shape, torch.Size((1, 1)))
  
file_path: pyro/tests/ops/test_special.py
imports: import pytest
  import torch
  from scipy.special import iv
  from torch import tensor
  from torch.autograd import grad
  from pyro.ops.special import get_quad_rule, log_beta, log_binomial, log_I1, safe_log
  from tests.common import assert_equal
label_desc: Use image node_id 5 for calling a global function with example usage: test_log_I1_shapes() without return types