code: def test_transform_bijection(batch_shape, event_shape):
      tf = OrderedTransform()
      assert tf.inv.inv is tf
      shape = torch.Size(batch_shape + event_shape)
      sample = Normal(0, 1).expand(shape).sample()
      tf_sample = tf(sample)
      inv_tf_sample = tf.inv(tf_sample)
      assert torch.allclose(sample, inv_tf_sample)
  
file_path: pyro/tests/distributions/test_ordered_logistic.py
imports: import pytest
  import torch
  from torch.autograd.functional import jacobian
  from pyro.distributions import Normal, OrderedLogistic
  from pyro.distributions.transforms import OrderedTransform
label_desc: Use image node_id 6 for calling a global function with example usage: test_transform_bijection(batch_shape, event_shape) without return types