Matthew Hoffman c976321541 Use variadic length tuple for torch.masked.DimOrDims (#149870) ... `tuple[int]` means only a tuple of length 1, which is not what was intended. ```python loss = torch.masked.mean(loss, mask=mask, dim=(-1, -2)) # Argument of type "tuple[Literal[-1], Literal[-2]]" cannot be assigned to parameter "dim" of type "DimOrDims" ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/149870 Approved by: https://github.com/Skylion007		2025-03-31 07:06:58 +00:00
..
maskedtensor	PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)	2025-01-20 22:37:26 +00:00
__init__.py	Add functions from torch.masked._ops to __all__ for torch.masked (#131288)	2024-08-01 05:45:08 +00:00
_docs.py	Fix global flake8 issues (#124771)	2024-04-26 15:35:53 +00:00
_ops.py	Use variadic length tuple for torch.masked.DimOrDims (#149870)	2025-03-31 07:06:58 +00:00