1 在Numpy 或Pytorch 中的網格上應用坐標掩碼. 作為強化學習問題的一部分,我需要節省gpu 內存。 網格的內存大小是二次方的(像素數)。 但是,只有代理所在的像素才是 ...
確定! 回上一頁