AffineGrid#
Domain:
ai.onnxSince 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:
[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).
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.
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
Node:
AffineGrid(theta, size) -> (grid)
Attributes:
align_corners = 0
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
Node:
AffineGrid(theta, size) -> (grid)
Attributes:
align_corners = 1
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
Node:
AffineGrid(theta, size) -> (grid)
Attributes:
align_corners = 0
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
Node:
AffineGrid(theta, size) -> (grid)
Attributes:
align_corners = 1
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]]]]]