def __getitem__(self, *args, **kwargs): m = Maze().generate(width=self.width, height=self.height) m = m._to_str_matrix(_np=True) m = torch.from_numpy(m) return m
def __iter__(self): m = Maze().generate(width=self.width, height=self.height) m = m._to_str_matrix(_np=True) m = torch.from_numpy(m) yield m