# tensorplay.nn.functional.affine_grid

Source: https://www.tensorplay.cn/docs/generated/tensorplay.nn.functional.affine_grid.html

# tensorplay.nn.functional.affine_grid

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

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

Port of aten/src/ATen/native/AffineGridGenerator.cpp.