Python segmented_tensor示例

示例#1

文件： segment_ops_test.py 项目： aaaqhbd/caffe2_1

 def test_sorted_segment_ops(self):
     SegmentsTester()._test(
         'SortedSegment',
         hu.segmented_tensor(dtype=np.float32,
                             is_sorted=True,
                             allow_empty=True),
         REFERENCES_ALL + REFERENCES_SORTED)(self)

示例#2

文件： segment_ops_test.py 项目： aaaqhbd/caffe2_1

 def test_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'UnsortedSegment',
         hu.segmented_tensor(dtype=np.float32,
                             is_sorted=False,
                             allow_empty=True),
         REFERENCES_ALL,
     )(self)

示例#3

文件： segment_ops_test.py 项目： gtgalone/pytorch

 def test_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'UnsortedSegment',
         hu.segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True
         ),
         REFERENCES_ALL,
     )(self)

示例#4

文件： segment_ops_test.py 项目： gtgalone/pytorch

 def test_sorted_segment_ops(self):
     SegmentsTester()._test(
         'SortedSegment',
         hu.segmented_tensor(
             dtype=np.float32,
             is_sorted=True,
             allow_empty=True
         ),
         REFERENCES_ALL + REFERENCES_SORTED
     )(self)

示例#5

文件： segment_ops_test.py 项目： aaaqhbd/caffe2_1

 def test_unsorted_segment_ops_gpu(self):
     SegmentsTester()._test(
         'UnsortedSegment',
         hu.segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True,
         ),
         REFERENCES_ALL,
         gpu=workspace.has_gpu_support,
         grad_check=False,
     )(self)

示例#6

文件： segment_ops_test.py 项目： gtgalone/pytorch

 def test_unsorted_segment_ops_gpu(self):
     SegmentsTester()._test(
         'UnsortedSegment',
         hu.segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True,
         ),
         REFERENCES_ALL,
         gpu=workspace.has_gpu_support,
         grad_check=False,
     )(self)

示例#7

文件： segment_ops_test.py 项目： zxsted/caffe2

 def test_unsorted_segment_ops(self):
     test('UnsortedSegment',
          hu.segmented_tensor(dtype=np.float32, is_sorted=False),
          REFERENCES_ALL)(self)

示例#8

文件： segment_ops_test.py 项目： zxsted/caffe2

 def test_sorted_segment_ops(self):
     test('SortedSegment',
          hu.segmented_tensor(dtype=np.float32, is_sorted=True),
          REFERENCES_ALL + REFERENCES_SORTED)(self)

示例#9

文件： segment_ops_test.py 项目： maniacs-satm/caffe2

 def test_unsorted_segment_ops(self):
     test(
         'UnsortedSegment',
         hu.segmented_tensor(dtype=np.float32, is_sorted=False),
         REFERENCES_ALL)(self)

示例#10

文件： segment_ops_test.py 项目： maniacs-satm/caffe2

 def test_sorted_segment_ops(self):
     test(
         'SortedSegment',
         hu.segmented_tensor(dtype=np.float32, is_sorted=True),
         REFERENCES_ALL + REFERENCES_SORTED)(self)