Skip to contents

Generates a 2D or 3D flow field (sampling grid), given a batch of affine matrices theta.


nnf_affine_grid(theta, size, align_corners = FALSE)



(Tensor) input batch of affine matrices with shape (\(N \times 2 \times 3\)) for 2D or (\(N \times 3 \times 4\)) for 3D


(torch.Size) the target output image size. (\(N \times C \times H \times W\) for 2D or \(N \times C \times D \times H \times W\) for 3D) Example: torch.Size((32, 3, 24, 24))


(bool, optional) if True, consider -1 and 1 to refer to the centers of the corner pixels rather than the image corners. Refer to nnf_grid_sample() for a more complete description. A grid generated by nnf_affine_grid() should be passed to nnf_grid_sample() with the same setting for this option. Default: False


This function is often used in conjunction with nnf_grid_sample() to build Spatial Transformer Networks_ .