mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Old implementation leads to incorrect output due to not handling the other batch sizes other than 3D tensors(B, M, N) Pull Request resolved: https://github.com/pytorch/pytorch/pull/144545 Approved by: https://github.com/malfet |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _masked.py | ||
| fft.py | ||
| linalg.py | ||
| nested.py | ||
| signal.py | ||
| sparse.py | ||
| special.py | ||