torch.ReplaceDim

export type ReplaceDim<
  S extends Shape,
  Dim extends number,
  NewSize extends number,
> = number extends S['length']
  ? DynamicShape
  : S extends readonly [infer First extends number, ...infer Rest extends number[]]
    ? Dim extends 0
      ? readonly [NewSize, ...Rest]
      : readonly [First, ...ReplaceDim<Rest, Subtract1<Dim>, NewSize>]
    : readonly [];

Sextends Shape

Dimextends number

NewSizeextends number

Replace dimension at index with new size. Used by narrow/slice operations.