tensorplay.nn.functional.affine_grid

tensorplay.nn.functional.affine_grid(theta: Tensor, size, align_corners=None) → Tensor[source]

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