code: def atleast_kd(x: ep.Tensor, k: int) -> ep.Tensor:
      shape = x.shape + (1,) * (k - x.ndim)
      return x.reshape(shape)
  
label_desc: Use image node_id 2 for calling a global function with example usage: atleast_kd(x, k) and returns: x