def read_chunk(self, idx, chunk_size, padding=(0, 0), nodes=None):
t_start, t_stop = self._get_t_start_t_stop(idx, chunk_size, padding)
local_shape = t_stop - t_start
if nodes is None:
nodes = numpy.arange(self.nb_channels, dtype=numpy.int32)
local_chunk = numpy.zeros((local_shape, len(nodes)), dtype=self.data_dtype)
for count, i in enumerate(nodes):
local_chunk[:, count] = self.data.get_entity(numpy.int64(i)).get_data(t_start, numpy.int64(local_shape))[0]
return self._scale_data_to_float32(local_chunk)
评论列表
文章目录