code: def _affine_grid_helper(draw):
      align_corners = draw(st.booleans())
      dims = draw(st.integers(4, 5))
      if dims == 4:
          size = draw(
              st.tuples(
                  st.integers(1, 20),
                  st.integers(1, 20),
                  st.integers(2, 20),
                  st.integers(2, 20),
              )
          )
          theta_dtype, theta = draw(
              helpers.dtype_and_values(
                  available_dtypes=helpers.get_dtypes("float"),
                  min_value=0,
                  max_value=1,
                  shape=(size[0], 2, 3),
              )
          )
          return theta_dtype, theta[0], size, align_corners
      else:
          size = draw(
              st.tuples(
                  st.integers(1, 20),
                  st.integers(1, 20),
                  st.integers(2, 20),
                  st.integers(2, 20),
                  st.integers(2, 20),
              )
          )
          theta_dtype, theta = draw(
              helpers.dtype_and_values(
                  available_dtypes=helpers.get_dtypes("float"),
                  min_value=0,
                  max_value=1,
                  shape=(size[0], 3, 4),
              )
          )
          return theta_dtype, theta[0], size, align_corners
  
file_path: ivy/ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_vision_functions.py
imports: from hypothesis import assume, strategies
  import ivy
  import ivy_tests.test_ivy.helpers
  from ivy_tests.test_ivy.helpers import handle_frontend_test
  from ivy_tests.test_ivy.test_functional.test_experimental.test_nn.test_layers import _interp_args
label_desc: Use image node_id 1 for calling a global function with example usage: _affine_grid_helper(draw) and returns: theta_dtype,  theta,  size,  align_corners, theta_dtype,  theta,  size,  align_corners