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