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

Add cummax

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/32169

Differential Revision: D19393236

Pulled By: anjali411

fbshipit-source-id: 5dac6b0a4038eb48458d4a0b253418daeccbb6bc
This commit is contained in:
anjali411
2020-01-14 16:36:56 -08:00
committed by Facebook Github Bot
parent 02c3493a84
commit 8dc67a014f
13 changed files with 187 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_namedtensor.py
View File

@ -977,6 +977,14 @@ class TestNamedTensor(TestCase):
for testcase, device in itertools.product(tests, torch.testing.get_all_device_types()):
_test(testcase, device=device)
def test_cummax(self):
for device in torch.testing.get_all_device_types():
names = ('N', 'D')
tensor = torch.rand(2, 3, names=names)
result = torch.cummax(tensor, 0)
self.assertEqual(result[0].names, names)
self.assertEqual(result[1].names, names)
def test_bitwise_not(self):
for device in torch.testing.get_all_device_types():
names = ('N', 'D')