mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
----
- torch.matmul -> CompositeImplicitAutograd -> dot_out (when left_dim == 1 & right_dim == 1)
-> mv_out (when left_dim == 2 & right_dim == 1)
-> mm_out (when left_dim == 1 & right_dim == 2)
-> ...
- torch.dot
- torch.vdot
- torch.mm
- torch.mv
ISSUE related:
https://github.com/pytorch/pytorch/issues/138399
Pull Request resolved: https://github.com/pytorch/pytorch/pull/148174
Approved by: https://github.com/jansel
|
||
|---|---|---|
| .. | ||
| __init__.py | ||
| decompositions_for_jvp.py | ||
| decompositions_for_rng.py | ||
| decompositions.py | ||