class_docstr: Jax pytree class that represents a ``ParametrizedHamiltonian``.
class_docstr_tok: ["Jax","pytree","class","that","represents","a","``","ParametrizedHamiltonian","``","."]
docstr: Function used by ``jax`` to unflatten the JaxParametrizedOperator.
  
  Args:
      param_coeffs (tuple): tuple containing the parametrized coefficients of the class
      matrices (tuple): tuple containing the matrices of the class
      reorder_fn(callable): callable or None indicating how parameters should be
          re-orderd to pass to the __call__ method
  
  Returns:
      JaxParametrizedOperator: a JaxParametrizedOperator instance
docstr_tok: ["Function","used","by","``","jax","``","to","unflatten","the","JaxParametrizedOperator",".","Args",":","param_coeffs","(","tuple",")",":","tuple","containing","the","parametrized","coefficients","of","the","class","matrices","(","tuple",")",":","tuple","containing","the","matrices","of","the","class","reorder_fn","(","callable",")",":","callable","or","None","indicating","how","parameters","should","be","re-orderd","to","pass","to","the","__call__","method","Returns",":","JaxParametrizedOperator",":","a","JaxParametrizedOperator","instance"]
code: def tree_unflatten(
      cls, param_coeffs: tuple, matrices: tuple, reorder_fn: callable
  ):
      """Function used by ``jax`` to unflatten the JaxParametrizedOperator.
  
      Args:
          param_coeffs (tuple): tuple containing the parametrized coefficients of the class
          matrices (tuple): tuple containing the matrices of the class
          reorder_fn(callable): callable or None indicating how parameters should be
              re-orderd to pass to the __call__ method
  
      Returns:
          JaxParametrizedOperator: a JaxParametrizedOperator instance
      """
      return cls(*matrices, param_coeffs, reorder_fn)
  
file_path: pennylane/pennylane/pulse/parametrized_hamiltonian_pytree.py
imports: from dataclasses import dataclass
  from typing import Callable, Optional, Tuple, Union
  import jax
  import jax.numpy
  from jax.experimental import sparse
  from jax.tree_util import register_pytree_node_class
  import pennylane
  from .parametrized_hamiltonian import ParametrizedHamiltonian
  from .hardware_hamiltonian import HardwareHamiltonian
label_desc: Use image node_id 4 for calling the ParametrizedHamiltonianPytree obj's underlying member method code with example usage: obj.tree_unflatten(cls, param_coeffs, matrices, reorder_fn) and returns: cls