සමානාත්මතාකාර්යය

19from typing import Tuple
20
21import torch
22from torch.utils.data import Dataset

25class ParityDataset(Dataset):

30    def __init__(self, n_samples: int, n_elems: int = 64):

35        self.n_samples = n_samples
36        self.n_elems = n_elems

38    def __len__(self):

42        return self.n_samples

44    def __getitem__(self, idx: int) -> Tuple[torch.Tensor, torch.Tensor]:

50        x = torch.zeros((self.n_elems,))

52        n_non_zero = torch.randint(1, self.n_elems + 1, (1,)).item()

54        x[:n_non_zero] = torch.randint(0, 2, (n_non_zero,)) * 2 - 1

56        x = x[torch.randperm(self.n_elems)]

59        y = (x == 1.).sum() % 2

62        return x, y