Able to learn x and y translation affine matrix