affine_grid.py 10.4 KB