torch.InsertDim

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

Sextends Shape

Dimextends number

Sizeextends number

Insert a dimension of given size at the specified position. InsertDim<[2, 3], 0, 4> = [4, 2, 3] InsertDim<[2, 3], 1, 4> = [2, 4, 3] InsertDim<[2, 3], 2, 4> = [2, 3, 4]