Method __getitem__

diff2flow/dataloader.py:277–281

(self, idx)

        return int(self.num_samples)

    def __getitem__(self, idx):
        return {
            k: v if isinstance(v, str) else torch.randn(v)
            for  k, v in self.kwargs.items()
        }


if __name__ == "__main__":