frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix tensor creation with empty names crash (#163957)

Browse Source 
Partially fixes #148324

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163957
Approved by: https://github.com/malfet, https://github.com/janeyx99
This commit is contained in:
arkadip-maitra
2025-09-26 17:40:55 +00:00
committed by PyTorch MergeBot
parent 9534c59311
commit d7491fb1c1
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_namedtensor.py
View File

@ -280,6 +280,11 @@ class TestNamedTensor(TestCase):
self.assertEqual(named_tensor.diagonal(outdim='E', dim1='B', dim2='D').names,
['A', 'C', 'E'])
def test_empty_names(self):
ref_tensor = torch.tensor([[1, 2, 3, 4], [4, 3, 2, 1]])
empty_named_tensor = torch.tensor([[1, 2, 3, 4], [4, 3, 2, 1]], names=[])
self.assertEqual(ref_tensor, empty_named_tensor)
def test_max_pooling(self):
def check_tuple_return(op, inputs, expected_names):
values, indices = op(*inputs)