Example of a torch dataset that load sparse matrix.

torch_dataset.py

class CustomImageDatasetNumpy(Dataset):
    def __init__(self, dir_path:str):
        self.dir_path = dir_path
        self.filenames = os.listdir(dir_path)

    def __len__(self):
        return len(self.filenames)

    def __getitem__(self, idx):
        np_path = self.dir_path + "/" + self.filenames[idx]
        sparse_array = sparse.load_npz(np_path)
        np_array = sparse_array.toarray()
        t = Tensor(np_array)
        return t, 0