>>> transform = ts.IndexTransform(input_rank=3)
>>> transform[ts.newaxis, 5]
Rank 3 -> 3 index space transform:
  Input domain:
    0: [0*, 1*)
    1: (-inf*, +inf*)
    2: (-inf*, +inf*)
  Output index maps:
    out[0] = 5
    out[1] = 0 + 1 * in[1]
    out[2] = 0 + 1 * in[2]