Does there exist an affine transform implementation that can take a transformation matrix and an image and backprop to both?