Trayvon
Trayvon
It should be `[move_curr // self.height, move_curr % self.height]`? As the shape of square_state is `(4, self.width, self.height)`, so the third dimension should be `height`? By the way, why reshape...