code: def _pytree_shape_dtype_struct(pytree):
      """Creates a shape structure that matches the types and shapes for the provided pytree."""
      leaves, struct = jax.tree_util.tree_flatten(pytree)
      new_leaves = [
          jax.ShapeDtypeStruct(jnp.shape(l), l.dtype) for l in leaves
      ]
      return jax.tree_util.tree_unflatten(struct, new_leaves)
  
file_path: pennylane/pennylane/workflow/interfaces/jax_jit.py
imports: import jax
  import jax.numpy
  import pennylane
  from pennylane.typing import ResultBatch
  from ..jacobian_products import _compute_jvps
  from .jax import _NonPytreeWrapper
label_desc: Use image node_id 7 for calling a global function with example usage: _pytree_shape_dtype_struct(pytree) and returns: jax