Exemplos de sparse_segmented_tensor em Python

Exemplo n.º 1

 def test_sparse_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True
         ),
         REFERENCES_ALL
     )(self)

Exemplo n.º 2

Arquivo: segment_ops_test.py Projeto: gtgalone/pytorch

 def test_sparse_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True
         ),
         REFERENCES_ALL
     )(self)

Exemplo n.º 3

Arquivo: segment_ops_test.py Projeto: zxsted/caffe2

 def test_sparse_unsorted_segment_ops(self):
     test('SparseUnsortedSegment',
          hu.sparse_segmented_tensor(dtype=np.float32, is_sorted=False),
          REFERENCES_ALL)(self)

Exemplo n.º 4

Arquivo: segment_ops_test.py Projeto: maniacs-satm/caffe2

 def test_sparse_unsorted_segment_ops(self):
     test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(dtype=np.float32, is_sorted=False),
         REFERENCES_ALL)(self)