torch.RemoveDim

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

Sextends Shape

Dimextends number

Remove a dimension from shape.