code: def eye_like(x: torch.Tensor, n: int) -> torch.Tensor:
      return (
          torch.eye(n, n, dtype=x.dtype, device=x.device)
          .unsqueeze(0)
          .repeat(x.shape[0], 1, 1)
      )
  
file_path: insightface/reconstruction/jmlr/losses.py
imports: import torch
  from torch import nn
  import torch.nn.functional
  import kornia
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: eye_like(x, n) and returns: torch