torch.TransposeDimsShape

export type TransposeDimsShape<
  S extends Shape,
  D0 extends number,
  D1 extends number,
> = number extends S['length']
  ? DynamicShape
  : D0 extends D1
    ? S
    : // Ensure D0 < D1 for consistent handling
      TupleOf<D0> extends [...TupleOf<D1>, ...infer _]
      ? TransposeDimsShapeImpl<S, D1, D0> // Swap order
      : TransposeDimsShapeImpl<S, D0, D1>;

Sextends Shape

D0extends number

D1extends number

Compute output shape of transpose(dim0, dim1).