class_docstr: Generate Jigsaw permutation indices for a batch of inputs.
  
  Args:
      grid: the Jigsaw puzzle grid. e.g. (2, 2) means
          each output will mix image patches in a 2x2 grid.
  
  Returns:
      A dict of parameters to be passed for transformation.
          - permutation (Tensor): Jigsaw permutation arrangement.
  
  Note:
      The generated random numbers are not reproducible across different devices and dtypes. By default,
      the parameters will be generated on CPU in float32. This can be changed by calling
      ``self.set_rng_device_and_dtype(device="cuda", dtype=torch.float64)``.
code: def __repr__(self) -> str:
      repr = f"grid={self.grid}"
      return repr
  
file_path: kornia/kornia/augmentation/random_generator/_2d/jigsaw.py
imports: from typing import Dict, Tuple
  import torch
  from kornia.augmentation.random_generator.base import RandomGeneratorBase
  from kornia.augmentation.random_generator.utils import randperm
  from kornia.augmentation.utils import _common_param_check
label_desc: Use image node_id 2 for calling the JigsawGenerator obj's underlying member method code with example usage: obj.__repr__() and returns: repr