code: def test_flow_shapes(
      bijection_func,
      batch_shape,
      event_shape,
  ):
      bijection = bijection_func(event_shape=event_shape)
      bijection.initialize()
      assert bijection.event_shape == event_shape
      x = nd.zeros(batch_shape + event_shape)
      y = bijection.f(x)
      y_inv = bijection.f_inv(x)
      assert y.shape == batch_shape + event_shape
      assert y_inv.shape == batch_shape + event_shape
  
file_path: gluonts/test/mx/distribution/test_flows.py
imports: from functools import partial
  import numpy
  import pytest
  import mxnet.ndarray
  from mxnet import autograd
  from gluonts.mx.distribution.iresnet import iresnet, log_abs_det
label_desc: Use image node_id 3 for calling a global function with example usage: test_flow_shapes(bijection_func, batch_shape, event_shape) without return types