code: def test_wrap_ufunc_out_shape_multi_input():
      out_shape = (2, 4)
      func = lambda x, y: np.random.rand(*out_shape)
      ary1 = np.ones((4, 100))
      ary2 = np.ones((4, 5))
      res = wrap_xarray_ufunc(
          func,
          ary1,
          ary2,
          func_kwargs={"out_shape": out_shape},
          ufunc_kwargs={"n_dims": 1},
      )
      assert res.shape == (*ary1.shape[:-1], *out_shape)
  
imports: import numpy
  import pytest
  from numpy.testing import assert_array_almost_equal
  from scipy.special import logsumexp
  from scipy.stats import circstd
  from ...data import from_dict, load_arviz_data
  from ...stats.density_utils import histogram
  from ...stats.stats_utils import ELPDData, _angle, _circfunc, _circular_standard_deviation, _sqrt, get_log_likelihood
  from ...stats.stats_utils import logsumexp
  from ...stats.stats_utils import make_ufunc, not_valid, stats_variance_2d, wrap_xarray_ufunc
label_desc: Use image node_id 5 for calling a global function with example usage: test_wrap_ufunc_out_shape_multi_input() without return types