torch.UnsqueezeShape

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

Sextends Shape

Dimextends number

Compute output shape of unsqueeze.