As part of a reinforcement learning problem, I need to save gpu memory.
The memory size of the grid is quadratic (in number of pixels). However, only the pixels where
