Method __init__

tests/cdslib/script/test_base_train.py:18–21

(self, xs: torch.Tensor, ys: torch.Tensor)

class QuickDataset(torch.utils.data.Dataset):
    def __init__(self, xs: torch.Tensor, ys: torch.Tensor):
        self.xs = xs
        self.ys = ys
        assert self.xs.size(0) == self.ys.size(0)

    def __len__(self):
        return self.xs.size(0)