Alan Du 50580b5053 Add minimal nn.functional.log_softmax support for NestedTensor (#159662) ... This only works for the jagged layout and for the non-batch and non-jagged dimensions. I did this mostly by copy-pasting from the existing softmax implementation, but it seems fairly straightforward and I think it should work. Pull Request resolved: https://github.com/pytorch/pytorch/pull/159662 Approved by: https://github.com/jbschlosser		2025-08-06 20:34:02 +00:00
..
_internal	Add minimal nn.functional.log_softmax support for NestedTensor (#159662)	2025-08-06 20:34:02 +00:00
__init__.py	Add check nested_tensor_from_jagged param jagged_dim >= 1 (#157770)	2025-07-10 00:34:39 +00:00