torch.AdaptivePool2dShape

export type AdaptivePool2dShape<
  S extends Shape,
  OutH extends number,
  OutW extends number,
> = S extends readonly [infer B extends number, infer C extends number, number, number]
  ? readonly [B, C, OutH, OutW]
  : S extends readonly [infer C extends number, number, number]
    ? readonly [C, OutH, OutW]
    : DynamicShape;

Sextends Shape

OutHextends number

OutWextends number

Output shape for adaptive pooling that targets a specific output size. Preserves batch and channel dimensions, uses specified target sizes.

Examples

// AdaptiveAvgPool2d with output_size (7, 7)
type Input = readonly [16, 512, 224, 224];
type Output = AdaptivePool2dShape<Input, 7, 7>;  // [16, 512, 7, 7]