.. _op_ai_onnx_AffineGrid: AffineGrid ========== - **Domain**: ``ai.onnx`` - **Since version**: 20 Generates a 2D or 3D flow field (sampling grid), given a batch of affine matrices theta (https://pytorch.org/docs/stable/generated/torch.nn.functional.affine_grid.html). An affine matrix ``theta`` is applied to a position tensor represented in its homogeneous expression. Here is an example in 3D: .. code-block:: [r00, r01, r02, t0] [x] [x'] [r10, r11, r12, t1] * [y] = [y'] [r20, r21, r22, t2] [z] [z'] [0, 0, 0, 1 ] [1] [1 ] where ``(x, y, z)`` is the position in the original space, ``(x', y', z')`` is the position in the output space. The last row is always ``[0, 0, 0, 1]`` and is not stored in the affine matrix. Therefore we have ``theta`` of shape ``(N, 2, 3)`` for 2D or ``(N, 3, 4)`` for 3D. Input ``size`` is used to define grid of positions evenly spaced in the original 2D or 3D space, with dimensions ranging from ``-1`` to ``1``. The output ``grid`` contains positions in the output space. When ``align_corners=1``, consider ``-1`` and ``1`` to refer to the centers of the corner pixels (mark ``v`` in illustration). .. code-block:: v v v v |-------------------|------------------| -1 0 1 When ``align_corners=0``, consider ``-1`` and ``1`` to refer to the outer edge of the corner pixels. .. code-block:: v v v v |------------------|-------------------| -1 0 1 **Inputs** - **theta** (*T1*): input batch of affine matrices with shape (N, 2, 3) for 2D or (N, 3, 4) for 3D - **size** (*T2*): the target output image size (N, C, H, W) for 2D or (N, C, D, H, W) for 3D **Outputs** - **grid** (*T1*): output tensor of shape (N, H, W, 2) of 2D sample coordinates or (N, D, H, W, 3) of 3D sample coordinates. **Attributes** - **align_corners** (*int*): if align_corners=1, consider -1 and 1 to refer to the centers of the corner pixels. if align_corners=0, consider -1 and 1 to refer to the outer edge the corner pixels. **Type Constraints** - **T1**: Constrain grid types to float tensors. Allowed types: tensor(bfloat16), tensor(double), tensor(float), tensor(float16). - **T2**: Constrain size's type to int64 tensors. Allowed types: tensor(int64). Examples -------- **test_affine_grid_2d** .. code-block:: text Node: AffineGrid(theta, size) -> (grid) Attributes: align_corners = 0 .. code-block:: text Inputs: theta: shape=(2, 2, 3), dtype=float32 [[[ 1.0889444 , -3.2880466 , 5. ], [ 2.0223253 , 1.0960155 , -3.3 ]], [[ 0.83578837, -0.55442286, 2.5 ], [ 0.78762794, 0.8397114 , 1.1 ]]] size: shape=(4,), dtype=int64 [2, 3, 5, 6] Outputs: grid: shape=(2, 5, 6, 2), dtype=float32 [[[[ 6.722984 , -5.8620834 ], [ 7.085965 , -5.187975 ], [ 7.448947 , -4.5138664 ], [ 7.811928 , -3.8397582 ], [ 8.17491 , -3.1656497 ], [ 8.537891 , -2.4915411 ]], [[ 5.407765 , -5.4236774 ], [ 5.7707467 , -4.749569 ], [ 6.133728 , -4.0754604 ], [ 6.4967093 , -3.401352 ], [ 6.8596907 , -2.7272434 ], [ 7.2226725 , -2.053135 ]], [[ 4.0925465 , -4.985271 ], [ 4.455528 , -4.3111625 ], [ 4.818509 , -3.637054 ], [ 5.181491 , -2.9629457 ], [ 5.544472 , -2.2888372 ], [ 5.9074535 , -1.6147287 ]], [[ 2.7773275 , -4.5468645 ], [ 3.140309 , -3.8727562 ], [ 3.5032907 , -3.198648 ], [ 3.866272 , -2.5245395 ], [ 4.2292533 , -1.850431 ], [ 4.592235 , -1.1763225 ]], [[ 1.4621091 , -4.1084585 ], [ 1.8250904 , -3.43435 ], [ 2.188072 , -2.7602417 ], [ 2.5510533 , -2.0861332 ], [ 2.9140348 , -1.4120247 ], [ 3.2770164 , -0.73791623]]], [[[ 2.247048 , -0.22812569], [ 2.525644 , 0.03441691], [ 2.8042402 , 0.29695958], [ 3.0828364 , 0.55950224], [ 3.3614326 , 0.8220449 ], [ 3.6400287 , 1.0845876 ]], [[ 2.0252788 , 0.10775888], [ 2.303875 , 0.37030149], [ 2.5824711 , 0.63284415], [ 2.8610673 , 0.8953868 ], [ 3.1396635 , 1.1579294 ], [ 3.4182596 , 1.4204721 ]], [[ 1.8035097 , 0.44364345], [ 2.0821059 , 0.70618606], [ 2.360702 , 0.9687287 ], [ 2.639298 , 1.2312714 ], [ 2.9178944 , 1.493814 ], [ 3.1964903 , 1.7563567 ]], [[ 1.5817406 , 0.779528 ], [ 1.8603367 , 1.0420706 ], [ 2.1389327 , 1.3046134 ], [ 2.4175289 , 1.567156 ], [ 2.696125 , 1.8296987 ], [ 2.9747212 , 2.0922413 ]], [[ 1.3599714 , 1.1154126 ], [ 1.6385676 , 1.3779552 ], [ 1.9171637 , 1.6404979 ], [ 2.1957598 , 1.9030405 ], [ 2.474356 , 2.1655831 ], [ 2.752952 , 2.4281259 ]]]] **test_affine_grid_2d_align_corners** .. code-block:: text Node: AffineGrid(theta, size) -> (grid) Attributes: align_corners = 1 .. code-block:: text Inputs: theta: shape=(2, 2, 3), dtype=float32 [[[ 1.0889444 , -3.2880466 , 5. ], [ 2.0223253 , 1.0960155 , -3.3 ]], [[ 0.83578837, -0.55442286, 2.5 ], [ 0.78762794, 0.8397114 , 1.1 ]]] size: shape=(4,), dtype=int64 [2, 3, 5, 6] Outputs: grid: shape=(2, 5, 6, 2), dtype=float32 [[[[ 7.1991024 , -6.4183407 ], [ 7.63468 , -5.6094103 ], [ 8.070257 , -4.8004804 ], [ 8.505836 , -3.9915502 ], [ 8.941414 , -3.18262 ], [ 9.376991 , -2.3736901 ]], [[ 5.555079 , -5.8703327 ], [ 5.990657 , -5.061403 ], [ 6.4262342 , -4.252473 ], [ 6.861812 , -3.4435425 ], [ 7.29739 , -2.6346126 ], [ 7.732968 , -1.8256824 ]], [[ 3.9110556 , -5.322325 ], [ 4.3466334 , -4.5133953 ], [ 4.7822113 , -3.704465 ], [ 5.2177887 , -2.8955348 ], [ 5.6533666 , -2.0866046 ], [ 6.0889444 , -1.2776747 ]], [[ 2.2670321 , -4.7743177 ], [ 2.70261 , -3.9653873 ], [ 3.138188 , -3.1564572 ], [ 3.5737658 , -2.347527 ], [ 4.009343 , -1.538597 ], [ 4.444921 , -0.72966695]], [[ 0.6230087 , -4.22631 ], [ 1.0585866 , -3.4173799 ], [ 1.4941645 , -2.6084495 ], [ 1.9297423 , -1.7995193 ], [ 2.3653202 , -0.9905894 ], [ 2.8008978 , -0.18165922]]], [[[ 2.2186346 , -0.52733934], [ 2.55295 , -0.21228814], [ 2.8872652 , 0.102763 ], [ 3.2215805 , 0.41781425], [ 3.5558958 , 0.73286533], [ 3.890211 , 1.0479165 ]], [[ 1.941423 , -0.10748363], [ 2.2757382 , 0.20756757], [ 2.6100538 , 0.5226187 ], [ 2.944369 , 0.83766997], [ 3.2786846 , 1.152721 ], [ 3.613 , 1.4677722 ]], [[ 1.6642116 , 0.3123721 ], [ 1.9985269 , 0.6274233 ], [ 2.3328424 , 0.9424744 ], [ 2.6671576 , 1.2575257 ], [ 3.001473 , 1.5725768 ], [ 3.3357882 , 1.887628 ]], [[ 1.3870002 , 0.7322278 ], [ 1.7213155 , 1.047279 ], [ 2.055631 , 1.3623302 ], [ 2.3899462 , 1.6773814 ], [ 2.7242618 , 1.9924325 ], [ 3.058577 , 2.3074837 ]], [[ 1.1097888 , 1.1520835 ], [ 1.4441041 , 1.4671347 ], [ 1.7784195 , 1.7821858 ], [ 2.1127348 , 2.097237 ], [ 2.44705 , 2.4122882 ], [ 2.7813654 , 2.7273393 ]]]] **test_affine_grid_3d** .. code-block:: text Node: AffineGrid(theta, size) -> (grid) Attributes: align_corners = 0 .. code-block:: text Inputs: theta: shape=(2, 3, 4), dtype=float32 [[[ 2.6830733 , -0.7943316 , 0.21829216, 5. ], [ 0.62225395, 3.2880466 , -0.53033006, -3.3 ], [ 0.24721935, 1.7241772 , 0.07809311, -1.1 ]], [[-0.35552558, 1.0044229 , 1.3995191 , 2.5 ], [ 0.17578839, 0.06028857, -0.9240381 , 1.1 ], [-1.1 , -0.45 , 0.45 , 2.2 ]]] size: shape=(5,), dtype=int64 [2, 3, 4, 5, 6] Outputs: grid: shape=(2, 4, 5, 6, 3), dtype=float32 [[[[[ 3.2358518 , -6.0512347 , -2.7439277 ], [ 4.13021 , -5.8438168 , -2.6615214 ], [ 5.0245676 , -5.636399 , -2.579115 ], [ 5.9189253 , -5.428981 , -2.4967084 ], [ 6.813283 , -5.221563 , -2.4143019 ], [ 7.7076406 , -5.014145 , -2.3318954 ]], [[ 2.9181192 , -4.7360163 , -2.054257 ], [ 3.812477 , -4.528598 , -1.9718504 ], [ 4.706835 , -4.3211803 , -1.889444 ], [ 5.6011925 , -4.113762 , -1.8070376 ], [ 6.49555 , -3.9063442 , -1.7246311 ], [ 7.389908 , -3.698926 , -1.6422246 ]], [[ 2.6003864 , -3.4207973 , -1.364586 ], [ 3.4947443 , -3.2133794 , -1.2821796 ], [ 4.389102 , -3.0059614 , -1.1997731 ], [ 5.2834597 , -2.7985435 , -1.1173667 ], [ 6.1778173 , -2.5911255 , -1.0349602 ], [ 7.0721755 , -2.3837075 , -0.9525537 ]], [[ 2.2826538 , -2.1055784 , -0.674915 ], [ 3.1770117 , -1.8981606 , -0.59250855], [ 4.0713696 , -1.6907426 , -0.5101021 ], [ 4.9657273 , -1.4833246 , -0.4276957 ], [ 5.860085 , -1.2759066 , -0.34528917], [ 6.754443 , -1.0684886 , -0.26288277]], [[ 1.964921 , -0.79036 , 0.01475596], [ 2.859279 , -0.582942 , 0.09716237], [ 3.7536368 , -0.37552404, 0.17956877], [ 4.6479945 , -0.16810608, 0.2619753 ], [ 5.542352 , 0.03931189, 0.3443817 ], [ 6.4367104 , 0.24673009, 0.4267882 ]]], [[[ 3.344998 , -6.3163996 , -2.7048812 ], [ 4.2393556 , -6.108982 , -2.6224747 ], [ 5.1337137 , -5.9015636 , -2.5400684 ], [ 6.0280714 , -5.694146 , -2.4576619 ], [ 6.922429 , -5.4867277 , -2.3752556 ], [ 7.816787 , -5.2793097 , -2.292849 ]], [[ 3.027265 , -5.001181 , -2.0152104 ], [ 3.9216232 , -4.793763 , -1.9328039 ], [ 4.815981 , -4.586345 , -1.8503973 ], [ 5.7103386 , -4.378927 , -1.767991 ], [ 6.6046963 , -4.1715093 , -1.6855845 ], [ 7.499054 , -3.964091 , -1.603178 ]], [[ 2.7095325 , -3.6859624 , -1.3255395 ], [ 3.6038904 , -3.4785445 , -1.243133 ], [ 4.498248 , -3.2711265 , -1.1607265 ], [ 5.392606 , -3.0637083 , -1.07832 ], [ 6.2869635 , -2.8562903 , -0.9959136 ], [ 7.1813216 , -2.6488724 , -0.91350716]], [[ 2.3918 , -2.3707438 , -0.63586843], [ 3.2861576 , -2.1633255 , -0.55346197], [ 4.1805153 , -1.9559076 , -0.4710555 ], [ 5.0748734 , -1.7484896 , -0.3886491 ], [ 5.969231 , -1.5410715 , -0.3062426 ], [ 6.863589 , -1.3336536 , -0.22383618]], [[ 2.074067 , -1.0555251 , 0.05380249], [ 2.968425 , -0.8481071 , 0.13620889], [ 3.862783 , -0.64068913, 0.2186153 ], [ 4.7571406 , -0.43327117, 0.3010218 ], [ 5.6514983 , -0.2258532 , 0.38342822], [ 6.5458565 , -0.018435 , 0.46583474]]], [[[ 3.454144 , -6.581565 , -2.6658347 ], [ 4.3485017 , -6.3741465 , -2.5834284 ], [ 5.2428594 , -6.166729 , -2.5010219 ], [ 6.1372175 , -5.9593105 , -2.4186153 ], [ 7.031575 , -5.751893 , -2.3362088 ], [ 7.925933 , -5.5444746 , -2.2538023 ]], [[ 3.1364112 , -5.266346 , -1.9761639 ], [ 4.0307693 , -5.058928 , -1.8937573 ], [ 4.925127 , -4.85151 , -1.8113508 ], [ 5.8194847 , -4.644092 , -1.7289444 ], [ 6.7138424 , -4.436674 , -1.646538 ], [ 7.6082 , -4.229256 , -1.5641315 ]], [[ 2.8186786 , -3.9511275 , -1.2864928 ], [ 3.7130365 , -3.7437093 , -1.2040864 ], [ 4.607394 , -3.5362914 , -1.12168 ], [ 5.501752 , -3.3288734 , -1.0392735 ], [ 6.3961096 , -3.1214554 , -0.95686704], [ 7.2904677 , -2.9140375 , -0.8744606 ]], [[ 2.500946 , -2.6359086 , -0.5968219 ], [ 3.3953037 , -2.4284906 , -0.51441544], [ 4.2896614 , -2.2210727 , -0.43200898], [ 5.184019 , -2.0136547 , -0.34960258], [ 6.0783772 , -1.8062366 , -0.26719606], [ 6.972735 , -1.5988187 , -0.18478966]], [[ 2.1832132 , -1.32069 , 0.09284902], [ 3.0775712 , -1.113272 , 0.17525542], [ 3.971929 , -0.905854 , 0.25766182], [ 4.8662868 , -0.698436 , 0.34006834], [ 5.7606444 , -0.49101806, 0.42247474], [ 6.6550026 , -0.28359985, 0.50488126]]], [[[ 3.56329 , -6.8467298 , -2.6267881 ], [ 4.457648 , -6.639312 , -2.5443816 ], [ 5.3520055 , -6.431894 , -2.4619753 ], [ 6.2463636 , -6.224476 , -2.3795688 ], [ 7.1407213 , -6.017058 , -2.2971625 ], [ 8.035079 , -5.80964 , -2.214756 ]], [[ 3.2455573 , -5.5315113 , -1.9371172 ], [ 4.1399155 , -5.324093 , -1.8547108 ], [ 5.034273 , -5.1166754 , -1.7723043 ], [ 5.928631 , -4.909257 , -1.6898978 ], [ 6.8229885 , -4.7018394 , -1.6074915 ], [ 7.717346 , -4.494421 , -1.525085 ]], [[ 2.9278247 , -4.2162924 , -1.2474463 ], [ 3.8221827 , -4.0088744 , -1.1650399 ], [ 4.7165403 , -3.8014565 , -1.0826334 ], [ 5.610898 , -3.5940385 , -1.000227 ], [ 6.5052557 , -3.3866205 , -0.9178205 ], [ 7.399614 , -3.1792026 , -0.83541405]], [[ 2.6100922 , -2.9010737 , -0.5577753 ], [ 3.5044498 , -2.6936555 , -0.47536886], [ 4.3988075 , -2.4862375 , -0.3929624 ], [ 5.293165 , -2.2788196 , -0.310556 ], [ 6.187523 , -2.0714016 , -0.22814947], [ 7.081881 , -1.8639836 , -0.14574307]], [[ 2.2923594 , -1.585855 , 0.13189554], [ 3.1867173 , -1.378437 , 0.21430194], [ 4.081075 , -1.1710191 , 0.29670835], [ 4.975433 , -0.9636011 , 0.37911487], [ 5.8697906 , -0.75618315, 0.46152127], [ 6.7641487 , -0.54876494, 0.5439278 ]]]], [[[[ 0.94309366, 1.5983074 , 3.1391668 ], [ 0.8245851 , 1.6569035 , 2.7725 ], [ 0.7060766 , 1.7154996 , 2.4058332 ], [ 0.58756804, 1.7740958 , 2.0391667 ], [ 0.46905947, 1.8326919 , 1.6725 ], [ 0.35055113, 1.891288 , 1.3058332 ]], [[ 1.3448628 , 1.6224228 , 2.9591668 ], [ 1.2263542 , 1.681019 , 2.5925 ], [ 1.1078458 , 1.7396151 , 2.2258334 ], [ 0.9893372 , 1.7982112 , 1.8591667 ], [ 0.8708286 , 1.8568074 , 1.4925001 ], [ 0.7523202 , 1.9154035 , 1.1258333 ]], [[ 1.746632 , 1.6465383 , 2.7791667 ], [ 1.6281234 , 1.7051344 , 2.4125 ], [ 1.509615 , 1.7637305 , 2.0458333 ], [ 1.3911064 , 1.8223267 , 1.6791668 ], [ 1.2725978 , 1.8809228 , 1.3125 ], [ 1.1540893 , 1.9395189 , 0.9458332 ]], [[ 2.1484013 , 1.6706537 , 2.5991669 ], [ 2.0298927 , 1.7292498 , 2.2325 ], [ 1.9113841 , 1.787846 , 1.8658333 ], [ 1.7928755 , 1.8464421 , 1.4991667 ], [ 1.6743671 , 1.9050382 , 1.1324999 ], [ 1.5558585 , 1.9636344 , 0.76583326]], [[ 2.5501704 , 1.6947691 , 2.4191668 ], [ 2.4316616 , 1.7533653 , 2.0525 ], [ 2.3131533 , 1.8119614 , 1.6858335 ], [ 2.1946447 , 1.8705575 , 1.3191667 ], [ 2.076136 , 1.9291537 , 0.9525 ], [ 1.9576277 , 1.9877498 , 0.5858333 ]]], [[[ 1.6428533 , 1.1362884 , 3.3641667 ], [ 1.5243447 , 1.1948845 , 2.9975 ], [ 1.4058361 , 1.2534807 , 2.6308334 ], [ 1.2873276 , 1.3120768 , 2.2641666 ], [ 1.1688191 , 1.370673 , 1.8975 ], [ 1.0503106 , 1.4292691 , 1.5308332 ]], [[ 2.0446224 , 1.1604038 , 3.1841667 ], [ 1.9261138 , 1.219 , 2.8175 ], [ 1.8076053 , 1.2775961 , 2.4508333 ], [ 1.6890968 , 1.3361921 , 2.0841668 ], [ 1.5705882 , 1.3947883 , 1.7175 ], [ 1.4520798 , 1.4533844 , 1.3508332 ]], [[ 2.4463916 , 1.1845193 , 3.0041666 ], [ 2.327883 , 1.2431154 , 2.6375 ], [ 2.2093744 , 1.3017116 , 2.2708333 ], [ 2.090866 , 1.3603077 , 1.9041667 ], [ 1.9723574 , 1.4189038 , 1.5374999 ], [ 1.8538489 , 1.4775 , 1.1708332 ]], [[ 2.8481607 , 1.2086346 , 2.8241668 ], [ 2.7296522 , 1.2672307 , 2.4575 ], [ 2.6111436 , 1.3258269 , 2.0908334 ], [ 2.4926353 , 1.384423 , 1.7241666 ], [ 2.3741267 , 1.4430192 , 1.3575 ], [ 2.255618 , 1.5016153 , 0.9908333 ]], [[ 3.24993 , 1.23275 , 2.6441667 ], [ 3.1314213 , 1.2913462 , 2.2775 ], [ 3.0129128 , 1.3499423 , 1.9108334 ], [ 2.8944042 , 1.4085385 , 1.5441667 ], [ 2.7758958 , 1.4671346 , 1.1775 ], [ 2.6573873 , 1.5257307 , 0.81083333]]], [[[ 2.3426127 , 0.6742693 , 3.5891666 ], [ 2.2241042 , 0.73286545, 3.2224998 ], [ 2.1055956 , 0.7914616 , 2.8558333 ], [ 1.9870871 , 0.8500577 , 2.4891667 ], [ 1.8685787 , 0.90865386, 2.1225 ], [ 1.7500701 , 0.96725 , 1.7558331 ]], [[ 2.744382 , 0.69838476, 3.4091668 ], [ 2.6258733 , 0.7569809 , 3.0425 ], [ 2.5073647 , 0.81557703, 2.6758332 ], [ 2.3888564 , 0.87417316, 2.3091667 ], [ 2.2703478 , 0.9327693 , 1.9425 ], [ 2.1518393 , 0.99136543, 1.5758333 ]], [[ 3.146151 , 0.7225002 , 3.2291667 ], [ 3.0276425 , 0.78109634, 2.8625 ], [ 2.909134 , 0.8396925 , 2.4958334 ], [ 2.7906256 , 0.89828855, 2.1291666 ], [ 2.672117 , 0.9568847 , 1.7625 ], [ 2.5536084 , 1.0154809 , 1.3958333 ]], [[ 3.5479202 , 0.7466156 , 3.0491667 ], [ 3.429412 , 0.8052117 , 2.6825 ], [ 3.310903 , 0.86380786, 2.3158333 ], [ 3.1923947 , 0.922404 , 1.9491668 ], [ 3.0738862 , 0.9810001 , 1.5825 ], [ 2.9553776 , 1.0395963 , 1.2158332 ]], [[ 3.9496894 , 0.77073103, 2.8691669 ], [ 3.8311808 , 0.82932717, 2.5025 ], [ 3.7126722 , 0.8879233 , 2.1358333 ], [ 3.594164 , 0.94651943, 1.7691667 ], [ 3.4756553 , 1.0051155 , 1.4024999 ], [ 3.3571467 , 1.0637116 , 1.0358332 ]]], [[[ 3.0423722 , 0.21225029, 3.8141665 ], [ 2.923864 , 0.27084643, 3.4475 ], [ 2.8053553 , 0.32944256, 3.0808334 ], [ 2.6868467 , 0.3880387 , 2.7141666 ], [ 2.5683382 , 0.44663483, 2.3474998 ], [ 2.4498296 , 0.5052309 , 1.9808332 ]], [[ 3.4441414 , 0.23636568, 3.6341667 ], [ 3.325633 , 0.2949618 , 3.2675 ], [ 3.2071245 , 0.35355794, 2.9008334 ], [ 3.088616 , 0.41215408, 2.5341666 ], [ 2.9701073 , 0.4707502 , 2.1675 ], [ 2.8515987 , 0.52934635, 1.8008332 ]], [[ 3.8459105 , 0.26048112, 3.4541667 ], [ 3.7274022 , 0.31907725, 3.0875 ], [ 3.6088936 , 0.3776734 , 2.7208333 ], [ 3.490385 , 0.43626952, 2.3541667 ], [ 3.3718765 , 0.49486566, 1.9875 ], [ 3.253368 , 0.5534618 , 1.6208332 ]], [[ 4.2476797 , 0.28459656, 3.2741666 ], [ 4.1291714 , 0.3431927 , 2.9075 ], [ 4.010663 , 0.40178883, 2.5408332 ], [ 3.8921542 , 0.46038496, 2.1741667 ], [ 3.7736459 , 0.5189811 , 1.8074999 ], [ 3.655137 , 0.57757723, 1.4408332 ]], [[ 4.649449 , 0.308712 , 3.0941668 ], [ 4.5309405 , 0.36730814, 2.7275 ], [ 4.4124317 , 0.42590427, 2.3608334 ], [ 4.2939234 , 0.4845004 , 1.9941666 ], [ 4.175415 , 0.54309654, 1.6274999 ], [ 4.056906 , 0.6016927 , 1.2608333 ]]]]] **test_affine_grid_3d_align_corners** .. code-block:: text Node: AffineGrid(theta, size) -> (grid) Attributes: align_corners = 1 .. code-block:: text Inputs: theta: shape=(2, 3, 4), dtype=float32 [[[ 2.6830733 , -0.7943316 , 0.21829216, 5. ], [ 0.62225395, 3.2880466 , -0.53033006, -3.3 ], [ 0.24721935, 1.7241772 , 0.07809311, -1.1 ]], [[-0.35552558, 1.0044229 , 1.3995191 , 2.5 ], [ 0.17578839, 0.06028857, -0.9240381 , 1.1 ], [-1.1 , -0.45 , 0.45 , 2.2 ]]] size: shape=(5,), dtype=int64 [2, 3, 4, 5, 6] Outputs: grid: shape=(2, 4, 5, 6, 3), dtype=float32 [[[[[ 2.89296603e+00, -6.67997074e+00, -3.14948988e+00], [ 3.96619534e+00, -6.43106842e+00, -3.05060196e+00], [ 5.03942490e+00, -6.18216705e+00, -2.95171404e+00], [ 6.11265421e+00, -5.93326569e+00, -2.85282660e+00], [ 7.18588352e+00, -5.68436432e+00, -2.75393867e+00], [ 8.25911236e+00, -5.43546295e+00, -2.65505123e+00]], [[ 2.49580026e+00, -5.03594732e+00, -2.28740120e+00], [ 3.56902957e+00, -4.78704548e+00, -2.18851328e+00], [ 4.64225912e+00, -4.53814411e+00, -2.08962560e+00], [ 5.71548843e+00, -4.28924227e+00, -1.99073792e+00], [ 6.78871775e+00, -4.04034090e+00, -1.89185023e+00], [ 7.86194706e+00, -3.79143906e+00, -1.79296231e+00]], [[ 2.09863448e+00, -3.39192390e+00, -1.42531252e+00], [ 3.17186379e+00, -3.14302230e+00, -1.32642472e+00], [ 4.24509335e+00, -2.89412069e+00, -1.22753704e+00], [ 5.31832266e+00, -2.64521909e+00, -1.12864923e+00], [ 6.39155197e+00, -2.39631748e+00, -1.02976155e+00], [ 7.46478081e+00, -2.14741588e+00, -9.30873752e-01]], [[ 1.70146871e+00, -1.74790061e+00, -5.63223898e-01], [ 2.77469778e+00, -1.49899900e+00, -4.64336097e-01], [ 3.84792757e+00, -1.25009751e+00, -3.65448356e-01], [ 4.92115688e+00, -1.00119567e+00, -2.66560614e-01], [ 5.99438620e+00, -7.52294302e-01, -1.67672932e-01], [ 7.06761551e+00, -5.03392458e-01, -6.87850714e-02]], [[ 1.30430293e+00, -1.03877306e-01, 2.98864722e-01], [ 2.37753201e+00, 1.45024538e-01, 3.97752404e-01], [ 3.45076180e+00, 3.93925905e-01, 4.96640325e-01], [ 4.52399111e+00, 6.42827749e-01, 5.95528007e-01], [ 5.59722042e+00, 8.91729116e-01, 6.94415689e-01], [ 6.67044973e+00, 1.14063048e+00, 7.93303370e-01]]], [[[ 3.03849411e+00, -7.03352356e+00, -3.09742761e+00], [ 4.11172342e+00, -6.78462219e+00, -2.99853992e+00], [ 5.18495274e+00, -6.53572083e+00, -2.89965200e+00], [ 6.25818253e+00, -6.28681898e+00, -2.80076456e+00], [ 7.33141136e+00, -6.03791761e+00, -2.70187664e+00], [ 8.40464020e+00, -5.78901577e+00, -2.60298896e+00]], [[ 2.64132833e+00, -5.38950062e+00, -2.23533916e+00], [ 3.71455765e+00, -5.14059925e+00, -2.13645124e+00], [ 4.78778696e+00, -4.89169741e+00, -2.03756356e+00], [ 5.86101627e+00, -4.64279556e+00, -1.93867576e+00], [ 6.93424606e+00, -4.39389420e+00, -1.83978808e+00], [ 8.00747490e+00, -4.14499283e+00, -1.74090028e+00]], [[ 2.24416256e+00, -3.74547720e+00, -1.37325048e+00], [ 3.31739187e+00, -3.49657559e+00, -1.27436268e+00], [ 4.39062119e+00, -3.24767399e+00, -1.17547488e+00], [ 5.46385098e+00, -2.99877238e+00, -1.07658720e+00], [ 6.53707981e+00, -2.74987078e+00, -9.77699459e-01], [ 7.61030912e+00, -2.50096941e+00, -8.78811717e-01]], [[ 1.84699678e+00, -2.10145402e+00, -5.11161804e-01], [ 2.92022586e+00, -1.85255241e+00, -4.12274003e-01], [ 3.99345541e+00, -1.60365081e+00, -3.13386261e-01], [ 5.06668472e+00, -1.35474920e+00, -2.14498520e-01], [ 6.13991451e+00, -1.10584760e+00, -1.15610838e-01], [ 7.21314335e+00, -8.56945992e-01, -1.67230368e-02]], [[ 1.44983101e+00, -4.57430601e-01, 3.50926876e-01], [ 2.52306008e+00, -2.08528996e-01, 4.49814558e-01], [ 3.59628963e+00, 4.03726101e-02, 5.48702359e-01], [ 4.66951895e+00, 2.89274216e-01, 6.47590041e-01], [ 5.74274826e+00, 5.38175583e-01, 7.46477842e-01], [ 6.81597757e+00, 7.87077188e-01, 8.45365524e-01]]], [[[ 3.18402243e+00, -7.38707733e+00, -3.04536557e+00], [ 4.25725174e+00, -7.13817549e+00, -2.94647789e+00], [ 5.33048105e+00, -6.88927364e+00, -2.84758997e+00], [ 6.40371037e+00, -6.64037228e+00, -2.74870253e+00], [ 7.47694016e+00, -6.39147091e+00, -2.64981461e+00], [ 8.55016899e+00, -6.14256954e+00, -2.55092692e+00]], [[ 2.78685665e+00, -5.74305391e+00, -2.18327713e+00], [ 3.86008573e+00, -5.49415207e+00, -2.08438921e+00], [ 4.93331528e+00, -5.24525070e+00, -1.98550153e+00], [ 6.00654459e+00, -4.99634933e+00, -1.88661373e+00], [ 7.07977390e+00, -4.74744749e+00, -1.78772604e+00], [ 8.15300369e+00, -4.49854565e+00, -1.68883824e+00]], [[ 2.38969088e+00, -4.09903049e+00, -1.32118833e+00], [ 3.46291995e+00, -3.85012913e+00, -1.22230065e+00], [ 4.53614950e+00, -3.60122752e+00, -1.12341285e+00], [ 5.60937881e+00, -3.35232592e+00, -1.02452517e+00], [ 6.68260813e+00, -3.10342431e+00, -9.25637364e-01], [ 7.75583744e+00, -2.85452271e+00, -8.26749623e-01]], [[ 1.99252510e+00, -2.45500731e+00, -4.59099770e-01], [ 3.06575394e+00, -2.20610571e+00, -3.60211968e-01], [ 4.13898373e+00, -1.95720410e+00, -2.61324227e-01], [ 5.21221304e+00, -1.70830250e+00, -1.62436485e-01], [ 6.28544235e+00, -1.45940089e+00, -6.35488033e-02], [ 7.35867167e+00, -1.21049929e+00, 3.53389978e-02]], [[ 1.59535933e+00, -8.10983896e-01, 4.02988911e-01], [ 2.66858840e+00, -5.62082291e-01, 5.01876593e-01], [ 3.74181795e+00, -3.13180685e-01, 6.00764394e-01], [ 4.81504726e+00, -6.42790794e-02, 6.99652076e-01], [ 5.88827658e+00, 1.84622288e-01, 7.98539877e-01], [ 6.96150589e+00, 4.33523893e-01, 8.97427559e-01]]], [[[ 3.32955027e+00, -7.74063015e+00, -2.99330330e+00], [ 4.40277958e+00, -7.49172878e+00, -2.89441586e+00], [ 5.47600937e+00, -7.24282742e+00, -2.79552794e+00], [ 6.54923868e+00, -6.99392605e+00, -2.69664049e+00], [ 7.62246799e+00, -6.74502468e+00, -2.59775257e+00], [ 8.69569683e+00, -6.49612236e+00, -2.49886465e+00]], [[ 2.93238473e+00, -6.09660721e+00, -2.13121510e+00], [ 4.00561380e+00, -5.84770584e+00, -2.03232718e+00], [ 5.07884359e+00, -5.59880447e+00, -1.93343937e+00], [ 6.15207291e+00, -5.34990215e+00, -1.83455157e+00], [ 7.22530222e+00, -5.10100079e+00, -1.73566389e+00], [ 8.29853153e+00, -4.85209942e+00, -1.63677621e+00]], [[ 2.53521895e+00, -4.45258427e+00, -1.26912630e+00], [ 3.60844803e+00, -4.20368242e+00, -1.17023849e+00], [ 4.68167782e+00, -3.95478082e+00, -1.07135081e+00], [ 5.75490713e+00, -3.70587921e+00, -9.72463012e-01], [ 6.82813644e+00, -3.45697761e+00, -8.73575330e-01], [ 7.90136528e+00, -3.20807600e+00, -7.74687529e-01]], [[ 2.13805318e+00, -2.80856085e+00, -4.07037675e-01], [ 3.21128225e+00, -2.55965900e+00, -3.08149874e-01], [ 4.28451157e+00, -2.31075764e+00, -2.09262133e-01], [ 5.35774136e+00, -2.06185579e+00, -1.10374391e-01], [ 6.43097019e+00, -1.81295431e+00, -1.14866495e-02], [ 7.50419998e+00, -1.56405270e+00, 8.74010324e-02]], [[ 1.74088740e+00, -1.16453719e+00, 4.55051064e-01], [ 2.81411648e+00, -9.15635824e-01, 5.53938746e-01], [ 3.88734579e+00, -6.66733980e-01, 6.52826428e-01], [ 4.96057510e+00, -4.17832613e-01, 7.51714110e-01], [ 6.03380489e+00, -1.68931246e-01, 8.50602031e-01], [ 7.10703373e+00, 7.99705982e-02, 9.49489713e-01]]]], [[[[ 4.51583624e-01, 1.78796124e+00, 3.29999995e+00], [ 3.09373379e-01, 1.85827661e+00, 2.86000013e+00], [ 1.67163134e-01, 1.92859197e+00, 2.42000008e+00], [ 2.49528885e-02, 1.99890733e+00, 1.98000002e+00], [-1.17257357e-01, 2.06922269e+00, 1.53999996e+00], [-2.59467602e-01, 2.13953781e+00, 1.10000002e+00]], [[ 9.53795075e-01, 1.81810546e+00, 3.07500005e+00], [ 8.11584830e-01, 1.88842082e+00, 2.63499999e+00], [ 6.69374585e-01, 1.95873618e+00, 2.19500017e+00], [ 5.27164340e-01, 2.02905154e+00, 1.75500000e+00], [ 3.84953976e-01, 2.09936690e+00, 1.31500006e+00], [ 2.42743969e-01, 2.16968226e+00, 8.75000000e-01]], [[ 1.45600653e+00, 1.84824967e+00, 2.85000014e+00], [ 1.31379628e+00, 1.91856503e+00, 2.41000009e+00], [ 1.17158604e+00, 1.98888040e+00, 1.97000003e+00], [ 1.02937579e+00, 2.05919576e+00, 1.52999997e+00], [ 8.87165546e-01, 2.12951112e+00, 1.09000003e+00], [ 7.44955301e-01, 2.19982648e+00, 6.50000095e-01]], [[ 1.95821798e+00, 1.87839401e+00, 2.62500000e+00], [ 1.81600773e+00, 1.94870937e+00, 2.18500018e+00], [ 1.67379749e+00, 2.01902485e+00, 1.74500012e+00], [ 1.53158724e+00, 2.08934021e+00, 1.30500007e+00], [ 1.38937700e+00, 2.15965557e+00, 8.65000010e-01], [ 1.24716675e+00, 2.22997093e+00, 4.24999952e-01]], [[ 2.46042943e+00, 1.90853834e+00, 2.40000010e+00], [ 2.31821918e+00, 1.97885370e+00, 1.96000004e+00], [ 2.17600894e+00, 2.04916906e+00, 1.52000010e+00], [ 2.03379869e+00, 2.11948442e+00, 1.07999992e+00], [ 1.89158845e+00, 2.18979979e+00, 6.40000105e-01], [ 1.74937820e+00, 2.26011515e+00, 2.00000048e-01]]], [[[ 1.38459635e+00, 1.17193580e+00, 3.59999990e+00], [ 1.24238610e+00, 1.24225116e+00, 3.16000009e+00], [ 1.10017586e+00, 1.31256652e+00, 2.72000003e+00], [ 9.57965612e-01, 1.38288188e+00, 2.27999997e+00], [ 8.15755367e-01, 1.45319724e+00, 1.84000003e+00], [ 6.73545122e-01, 1.52351260e+00, 1.40000010e+00]], [[ 1.88680780e+00, 1.20208001e+00, 3.37500000e+00], [ 1.74459755e+00, 1.27239537e+00, 2.93499994e+00], [ 1.60238731e+00, 1.34271073e+00, 2.49500012e+00], [ 1.46017706e+00, 1.41302609e+00, 2.05500007e+00], [ 1.31796682e+00, 1.48334146e+00, 1.61500001e+00], [ 1.17575657e+00, 1.55365682e+00, 1.17500007e+00]], [[ 2.38901925e+00, 1.23222435e+00, 3.15000010e+00], [ 2.24680901e+00, 1.30253971e+00, 2.71000004e+00], [ 2.10459876e+00, 1.37285507e+00, 2.26999998e+00], [ 1.96238852e+00, 1.44317043e+00, 1.82999992e+00], [ 1.82017827e+00, 1.51348579e+00, 1.39000010e+00], [ 1.67796803e+00, 1.58380103e+00, 9.50000048e-01]], [[ 2.89123082e+00, 1.26236868e+00, 2.92500019e+00], [ 2.74902058e+00, 1.33268392e+00, 2.48500013e+00], [ 2.60681033e+00, 1.40299928e+00, 2.04500008e+00], [ 2.46460009e+00, 1.47331476e+00, 1.60500002e+00], [ 2.32238984e+00, 1.54363000e+00, 1.16500008e+00], [ 2.18017960e+00, 1.61394536e+00, 7.25000024e-01]], [[ 3.39344215e+00, 1.29251289e+00, 2.70000005e+00], [ 3.25123191e+00, 1.36282825e+00, 2.25999999e+00], [ 3.10902166e+00, 1.43314362e+00, 1.82000005e+00], [ 2.96681142e+00, 1.50345898e+00, 1.38000000e+00], [ 2.82460117e+00, 1.57377434e+00, 9.40000057e-01], [ 2.68239093e+00, 1.64408970e+00, 5.00000119e-01]]], [[[ 2.31760907e+00, 5.55910349e-01, 3.90000010e+00], [ 2.17539883e+00, 6.26225650e-01, 3.46000004e+00], [ 2.03318858e+00, 6.96541011e-01, 3.01999998e+00], [ 1.89097834e+00, 7.66856372e-01, 2.57999992e+00], [ 1.74876809e+00, 8.37171733e-01, 2.14000010e+00], [ 1.60655785e+00, 9.07487094e-01, 1.70000005e+00]], [[ 2.81982064e+00, 5.86054564e-01, 3.67500019e+00], [ 2.67761040e+00, 6.56369925e-01, 3.23500013e+00], [ 2.53540015e+00, 7.26685286e-01, 2.79500008e+00], [ 2.39318991e+00, 7.97000647e-01, 2.35500002e+00], [ 2.25097966e+00, 8.67316008e-01, 1.91499996e+00], [ 2.10876942e+00, 9.37631369e-01, 1.47500014e+00]], [[ 3.32203197e+00, 6.16198897e-01, 3.45000005e+00], [ 3.17982173e+00, 6.86514258e-01, 3.01000023e+00], [ 3.03761148e+00, 7.56829619e-01, 2.57000017e+00], [ 2.89540124e+00, 8.27144980e-01, 2.13000011e+00], [ 2.75319099e+00, 8.97460341e-01, 1.69000006e+00], [ 2.61098075e+00, 9.67775643e-01, 1.25000000e+00]], [[ 3.82424355e+00, 6.46343172e-01, 3.22499990e+00], [ 3.68203330e+00, 7.16658533e-01, 2.78500009e+00], [ 3.53982306e+00, 7.86973834e-01, 2.34500003e+00], [ 3.39761281e+00, 8.57289255e-01, 1.90499997e+00], [ 3.25540257e+00, 9.27604556e-01, 1.46500015e+00], [ 3.11319232e+00, 9.97919917e-01, 1.02499998e+00]], [[ 4.32645512e+00, 6.76487446e-01, 3.00000000e+00], [ 4.18424463e+00, 7.46802807e-01, 2.56000018e+00], [ 4.04203415e+00, 8.17118168e-01, 2.12000012e+00], [ 3.89982414e+00, 8.87433529e-01, 1.68000007e+00], [ 3.75761390e+00, 9.57748890e-01, 1.24000001e+00], [ 3.61540365e+00, 1.02806425e+00, 8.00000072e-01]]], [[[ 3.25062180e+00, -6.01149797e-02, 4.19999981e+00], [ 3.10841155e+00, 1.02002621e-02, 3.75999999e+00], [ 2.96620131e+00, 8.05156231e-02, 3.31999993e+00], [ 2.82399106e+00, 1.50831044e-01, 2.88000011e+00], [ 2.68178082e+00, 2.21146405e-01, 2.44000006e+00], [ 2.53957057e+00, 2.91461706e-01, 2.00000000e+00]], [[ 3.75283337e+00, -2.99707651e-02, 3.97500014e+00], [ 3.61062288e+00, 4.03445959e-02, 3.53500009e+00], [ 3.46841288e+00, 1.10659957e-01, 3.09500003e+00], [ 3.32620239e+00, 1.80975318e-01, 2.65499997e+00], [ 3.18399239e+00, 2.51290679e-01, 2.21499991e+00], [ 3.04178190e+00, 3.21606040e-01, 1.77500010e+00]], [[ 4.25504494e+00, 1.73568726e-04, 3.75000000e+00], [ 4.11283445e+00, 7.04889297e-02, 3.30999994e+00], [ 3.97062421e+00, 1.40804231e-01, 2.86999989e+00], [ 3.82841396e+00, 2.11119592e-01, 2.43000007e+00], [ 3.68620372e+00, 2.81434953e-01, 1.99000001e+00], [ 3.54399347e+00, 3.51750314e-01, 1.54999995e+00]], [[ 4.75725603e+00, 3.03177834e-02, 3.52500010e+00], [ 4.61504602e+00, 1.00633144e-01, 3.08500004e+00], [ 4.47283554e+00, 1.70948505e-01, 2.64499998e+00], [ 4.33062553e+00, 2.41263866e-01, 2.20499992e+00], [ 4.18841505e+00, 3.11579227e-01, 1.76500010e+00], [ 4.04620504e+00, 3.81894588e-01, 1.32500005e+00]], [[ 5.25946760e+00, 6.04621172e-02, 3.30000019e+00], [ 5.11725712e+00, 1.30777419e-01, 2.86000013e+00], [ 4.97504711e+00, 2.01092780e-01, 2.42000008e+00], [ 4.83283710e+00, 2.71408141e-01, 1.98000002e+00], [ 4.69062662e+00, 3.41723502e-01, 1.53999996e+00], [ 4.54841614e+00, 4.12038863e-01, 1.10000014e+00]]]]]