code: def _new_eye(x, shape):
      return torch.eye(shape[-1]).expand(shape + (-1,))
  
imports: import numbers
  import typing
  import torch
  import funsor.ops
label_desc: Use image node_id 37 for calling a global function with example usage: _new_eye(x, shape) and returns: torch