code: def test_jax_shape_ops():
      x_np = np.zeros((20, 3))
      x = Shape()(pt.as_tensor_variable(x_np))
      x_fg = FunctionGraph([], [x])
  
      compare_jax_and_py(x_fg, [], must_be_device_array=False)
  
      x = Shape_i(1)(pt.as_tensor_variable(x_np))
      x_fg = FunctionGraph([], [x])
  
      compare_jax_and_py(x_fg, [], must_be_device_array=False)
  
file_path: pytensor/tests/link/jax/test_shape.py
imports: import numpy
  import pytest
  import pytensor.tensor
  from pytensor.compile.ops import DeepCopyOp, ViewOp
  from pytensor.configdefaults import config
  from pytensor.graph.fg import FunctionGraph
  from pytensor.tensor.shape import Shape, Shape_i, Unbroadcast, reshape
  from pytensor.tensor.type import iscalar, vector
  from tests.link.jax.test_basic import compare_jax_and_py
label_desc: Use image node_id 1 for calling a global function with example usage: test_jax_shape_ops() without return types