pytorch/torch/nested
David Berard 278a0e1600 [NestedTensor] Support binary pointwise ops with >2 inputs (if inputs are non-tensors) (#119419)
It should usually be safe to run pointwise binary ops with >2 inputs. e.g. threshold_backward(tensor, tensor, scalar): we just operate on the values of the nested tensors, and pass in the other args as-is.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/119419
Approved by: https://github.com/soulitzer
2024-02-08 20:06:40 +00:00
..
_internal [NestedTensor] Support binary pointwise ops with >2 inputs (if inputs are non-tensors) (#119419) 2024-02-08 20:06:40 +00:00
__init__.py Fix return type hint for list types (#118238) 2024-01-25 23:35:20 +00:00