torch.ValidatedRearrangeShape

export type ValidatedRearrangeShape<
  Pattern extends string,
  InputShape extends Shape,
  Axes extends AxesRecord = Record<string, never>,
> =
  RearrangeShape<Pattern, InputShape, Axes> extends infer R
    ? R extends { readonly __isShapeError: true }
      ? R & readonly number[]
      : R extends readonly number[]
        ? R
        : DynamicShape
    : DynamicShape;

Patternextends string

InputShapeextends Shape

Axesextends AxesRecord

Validated shape that always extends readonly number[]. Returns computed shape on success, error type on failure.