fn
affine_grid_op
→TensorImplPtrint affine_grid_op(const int & theta, int N, int H, int W, bool align_corners)Python wrappers
Public Python APIs implemented by this engine symbol.Public affine-grid entry point.
Thin wrapper that delegates to AffineGridBackward::forward.
Parameters
thetaTensorImplPtrAffine matrices of shape
(N, 2, 3).NintBatch size (must equal
theta.shape()[0]).HintOutput grid height.
WintOutput grid width.
align_cornersboolLattice placement mode.
Returns
TensorImplPtrSampling grid of shape (N, H, W, 2).