tensorplay.nn.functional.affine_grid

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

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