torch.NegativeDim

export type NegativeDim<
  D extends number,
  Ndim extends number,
> = `${D}` extends `-${infer Abs extends number}`
  ? Subtract1<Add<Ndim, 1>> extends infer Result extends number
    ? Add<Result, Subtract1<Abs>> extends infer Final extends number
      ? Final
      : number
    : number
  : D;

Dextends number

Ndimextends number

Convert negative dimension to positive. NegativeDim<-1, 3> = 2 (last dim of 3D tensor)