torch.fx.experimental.merge_matmul.merge_matmul

torch.fx.experimental.merge_matmul.merge_matmul(in_mod) [исходный код]

Преобразование графа, объединяющее операции умножения матриц с одинаковым правым операндом в одну большую операцию умножения матриц.

           ____      _________        _________
  ----    |    |    |         |     M|  A * C  |
M| A  |  T| B  | * K|    C    | =    |---------|
  ---- ,  |    |    |         |     T|  B * C  |
   K       ----      ---------        ---------
            K            R                R

Тип возвращаемого значения:

GraphModule