pytorch/torch/nested
Joel Schlosser 991f8f8ec3 Bias gradient calculation for NJT linear backward (#136660)
Previously NYI - @mikaylagawarecki needs it for Transformers.

Fixes #136652
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136660
Approved by: https://github.com/soulitzer
2024-09-26 21:38:10 +00:00
..
_internal Bias gradient calculation for NJT linear backward (#136660) 2024-09-26 21:38:10 +00:00
__init__.py Implement 2D version of masked_select for nestedtensors (#133889) 2024-08-20 21:46:32 +00:00