Method fetch

python/oneflow/utils/data/_utils/fetch.py:63–68

(self, possibly_batched_index)

        )

    def fetch(self, possibly_batched_index):
        if self.auto_collation:
            data = [self.dataset[idx] for idx in possibly_batched_index]
        else:
            data = self.dataset[possibly_batched_index]
        return self.collate_fn(data)