it is line 43 of the AffineGridGenerator.cpp, it has a condition that makes sure that the third component of the theta variable is equal to 3 where i want to create a theta which only has its third component equal to 2 since im only capturing simple translations and thus dont need the full affine transformation.