b = a + 1 + self.len_prompt
sequences = torch.empty(nb, b + 1 + self.len_result, dtype=torch.int64)
nb_operators = torch.randint(self.operators.size(0), (nb,))
b = a + 1 + self.len_prompt
sequences = torch.empty(nb, b + 1 + self.len_result, dtype=torch.int64)
nb_operators = torch.randint(self.operators.size(0), (nb,))