torch.SqueezeShape

export type SqueezeShape<
  S extends Shape,
  Dim extends number | undefined,
> = number extends S['length']
  ? DynamicShape
  : Dim extends undefined
    ? FilterOnes<S>
    : Dim extends number
      ? SqueezeAt<S, Dim>
      : DynamicShape;

Sextends Shape

Dimextends number | undefined

Compute output shape of squeeze.