I am trying to use torch.nn.function.affine_grid and grid sample functionalities for a Spatial Transformer.  After the affine transformation the the image has black regions near the edges due to the shifting and rotation of the original image. Is there some way to define the fill value instead of having a default black fill in the empty regions. (like in the attatched image)