Python sparse_splice_offset_numpy Examples

Example #1

File: Dataset.py Project: ZhangAustin/returnn

 def _get_data_slice_sparse(self, seq_idx, key, start_frame, end_frame):
     key_prefix = key[:key.index("[")]
     sparse_info = key[key.index("[") + 1:key.index("]")].split(":")
     assert len(sparse_info) == 4
     assert tuple(sparse_info[0:3]) == ("sparse", "coo", "2")
     s0 = self.get_data(seq_idx, "%s[sparse:coo:2:0]" % key_prefix)
     assert s0 is not None
     from NativeOp import sparse_splice_offset_numpy
     s0_start = sparse_splice_offset_numpy(s0, start_frame)
     s0_end = sparse_splice_offset_numpy(s0, end_frame)
     if sparse_info[-1] == "0":
         return s0[s0_start:s0_end] - start_frame
     else:
         data = self.get_data(seq_idx, key)
         return data[s0_start:s0_end]

Example #2

File: Dataset.py Project: atuxhe/returnn

 def _get_data_slice_sparse(self, seq_idx, key, start_frame, end_frame):
   key_prefix = key[:key.index("[")]
   sparse_info = key[key.index("[") + 1:key.index("]")].split(":")
   assert len(sparse_info) == 4
   assert tuple(sparse_info[0:3]) == ("sparse", "coo", "2")
   s0 = self.get_data(seq_idx, "%s[sparse:coo:2:0]" % key_prefix)
   assert s0 is not None
   from NativeOp import sparse_splice_offset_numpy
   s0_start = sparse_splice_offset_numpy(s0, start_frame)
   s0_end = sparse_splice_offset_numpy(s0, end_frame)
   if sparse_info[-1] == "0":
     return s0[s0_start:s0_end] - start_frame
   else:
     data = self.get_data(seq_idx, key)
     return data[s0_start:s0_end]