class_docstr: AugmentationBase3D base class for customized augmentation implementations.
  
  Args:
      p: probability for applying an augmentation. This param controls the augmentation probabilities
        element-wise for a batch.
      p_batch: probability for applying an augmentation to a batch. This param controls the augmentation
        probabilities batch-wise.
      same_on_batch: apply the same transformation across the batch.
docstr: Return 4x4 identity matrix.
docstr_tok: ["Return","4x4","identity","matrix","."]
code: def identity_matrix(self, input: Tensor) -> Tensor:
      """Return 4x4 identity matrix."""
      return kornia.eye_like(4, input)
  
imports: from typing import Any, Dict, Optional
  from torch import float16, float32, float64
  import kornia
  from kornia.augmentation.base import _AugmentationBase
  from kornia.augmentation.utils import _transform_input3d, _validate_input_dtype
  from kornia.core import Tensor
  from kornia.geometry.boxes import Boxes3D
  from kornia.geometry.keypoints import Keypoints3D
label_desc: Use image node_id 3 for calling the AugmentationBase3D obj's underlying member method code with example usage: obj.identity_matrix(input) and returns: kornia