pytorch/torch/nested
Joel Schlosser 67e6c76a18 Support apply_(callable) sugar for CPU NJTs (#125416)
Example:
```python
nt.apply_(lambda x: x * 2)
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/125416
Approved by: https://github.com/soulitzer
2024-06-12 20:30:57 +00:00
..
_internal Support apply_(callable) sugar for CPU NJTs (#125416) 2024-06-12 20:30:57 +00:00
__init__.py Flip default value for mypy disallow_untyped_defs [7/11] (#127844) 2024-06-08 18:49:45 +00:00