code: def arange_at_position(xp, n_axes, axis, axis_len, device=None):
      x = xp.arange(axis_len, dtype=xp.int64, device=device)
      shape = [1] * n_axes
      shape[axis] = axis_len
      x = xp.reshape(x, shape)
      return x
  
file_path: einops/einops/experimental/indexing.py
imports: from typing import List, Union, TypeVar, Tuple
  from einops import EinopsError
label_desc: Use image node_id 1 for calling a global function with example usage: arange_at_position(xp, n_axes, axis, axis_len, device) and returns: x