torch.Matmul2DShape

export type Matmul2DShape<A extends Shape, B extends Shape> = A extends readonly [
  infer M extends number,
  infer K1 extends number,
]
  ? B extends readonly [infer K2 extends number, infer N extends number]
    ? K1 extends K2
      ? readonly [M, N]
      : matmul_error_inner_dimensions_do_not_match<K1, K2, A, B>
    : DynamicShape
  : DynamicShape;

Aextends Shape

Bextends Shape

Compute output shape of 2D matrix multiplication (strict).