ArgMax#
Domain:
ai.onnxSince version: 13
Computes the indices of the max elements of the input tensor’s element along the provided axis. The resulting tensor has the same rank as the input if keepdims equals 1. If keepdims equals 0, then the resulting tensor has the reduced dimension pruned. If select_last_index is True (default False), the index of the last occurrence of the max is selected if the max appears more than once in the input. Otherwise the index of the first occurrence is selected. The type of the output tensor is integer.
Inputs
data (T): An input tensor.
Outputs
reduced (tensor(int64)): Reduced output tensor with integer data type.
Type Constraints
T: Constrain input and output types to all numeric tensors. Allowed types: tensor(bfloat16), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8).
Examples#
test_cc_argmax_default_axis_example
Node:
ArgMax(data) -> (result)
Attributes:
keepdims = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(1, 2), dtype=int64
[[1, 1]]
test_cc_argmax_default_axis_example_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(1, 2), dtype=int64
[[1, 1]]
test_cc_argmax_default_axis_random
Node:
ArgMax(data) -> (result)
Attributes:
keepdims = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(1, 4, 5, 6), dtype=int64
[[[[0, 0, 2, 1, 2, 2],
[1, 0, 2, 0, 2, 1],
[1, 2, 0, 2, 1, 0],
[1, 1, 1, 1, 0, 0],
[2, 1, 0, 2, 1, 1]],
[[2, 1, 1, 2, 1, 2],
[1, 1, 2, 0, 0, 0],
[2, 2, 0, 0, 1, 1],
[1, 1, 1, 2, 2, 2],
[1, 0, 1, 1, 2, 2]],
[[0, 2, 2, 0, 1, 0],
[0, 2, 2, 1, 0, 2],
[1, 1, 0, 0, 1, 1],
[0, 0, 0, 2, 1, 2],
[1, 0, 2, 0, 0, 1]],
[[1, 0, 1, 0, 1, 0],
[2, 2, 0, 1, 2, 2],
[1, 0, 2, 0, 0, 0],
[2, 0, 0, 0, 0, 1],
[1, 2, 2, 2, 0, 1]]]]
test_cc_argmax_default_axis_random_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(1, 4, 5, 6), dtype=int64
[[[[0, 0, 2, 1, 2, 2],
[1, 0, 2, 0, 2, 1],
[1, 2, 0, 2, 1, 0],
[1, 1, 1, 1, 0, 0],
[2, 1, 0, 2, 1, 1]],
[[2, 1, 1, 2, 1, 2],
[1, 1, 2, 0, 0, 0],
[2, 2, 0, 0, 1, 1],
[1, 1, 1, 2, 2, 2],
[1, 0, 1, 1, 2, 2]],
[[0, 2, 2, 0, 1, 0],
[0, 2, 2, 1, 0, 2],
[1, 1, 0, 0, 1, 1],
[0, 0, 0, 2, 1, 2],
[1, 0, 2, 0, 0, 1]],
[[1, 0, 1, 0, 1, 0],
[2, 2, 0, 1, 2, 2],
[1, 0, 2, 0, 0, 0],
[2, 0, 0, 0, 0, 1],
[1, 2, 2, 2, 0, 1]]]]
test_cc_argmax_keepdims_example
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2, 1), dtype=int64
[[0],
[1]]
test_cc_argmax_keepdims_example_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2, 1), dtype=int64
[[1],
[1]]
test_cc_argmax_keepdims_random
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 1, 5, 6), dtype=int64
[[[[0, 0, 2, 2, 3, 2],
[2, 0, 3, 0, 3, 1],
[3, 3, 1, 2, 3, 0],
[2, 2, 2, 3, 0, 0],
[1, 1, 0, 2, 3, 2]]],
[[[3, 1, 1, 0, 2, 0],
[1, 1, 0, 2, 0, 0],
[3, 2, 0, 3, 1, 2],
[0, 0, 0, 3, 0, 3],
[1, 0, 1, 1, 0, 2]]],
[[[1, 2, 2, 0, 0, 1],
[2, 2, 0, 0, 3, 0],
[1, 1, 2, 0, 1, 0],
[3, 1, 2, 1, 0, 2],
[0, 3, 3, 0, 1, 1]]]]
test_cc_argmax_keepdims_random_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 1, 5, 6), dtype=int64
[[[[0, 0, 2, 2, 3, 2],
[2, 0, 3, 0, 3, 1],
[3, 3, 1, 2, 3, 0],
[2, 2, 2, 3, 0, 0],
[1, 1, 0, 2, 3, 2]]],
[[[3, 1, 1, 0, 2, 0],
[1, 1, 0, 2, 0, 0],
[3, 2, 0, 3, 1, 2],
[0, 0, 0, 3, 0, 3],
[1, 0, 1, 1, 0, 2]]],
[[[1, 2, 2, 0, 0, 1],
[2, 2, 0, 0, 3, 0],
[1, 1, 2, 0, 1, 0],
[3, 1, 2, 1, 0, 2],
[0, 3, 3, 0, 1, 1]]]]
test_cc_argmax_negative_axis_keepdims_example
Node:
ArgMax(data) -> (result)
Attributes:
axis = -1
keepdims = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2, 1), dtype=int64
[[0],
[1]]
test_cc_argmax_negative_axis_keepdims_example_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = -1
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2, 1), dtype=int64
[[1],
[1]]
test_cc_argmax_negative_axis_keepdims_random
Node:
ArgMax(data) -> (result)
Attributes:
axis = -1
keepdims = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 4, 5, 1), dtype=int64
[[[[0],
[1],
[2],
[4],
[2]],
[[0],
[5],
[2],
[4],
[4]],
[[3],
[0],
[3],
[1],
[3]],
[[3],
[2],
[4],
[3],
[4]]],
[[[5],
[5],
[0],
[1],
[1]],
[[1],
[1],
[4],
[1],
[0]],
[[4],
[3],
[5],
[5],
[5]],
[[0],
[0],
[0],
[5],
[5]]],
[[[5],
[4],
[1],
[4],
[3]],
[[5],
[5],
[1],
[5],
[4]],
[[1],
[0],
[2],
[5],
[3]],
[[4],
[4],
[2],
[5],
[2]]]]
test_cc_argmax_negative_axis_keepdims_random_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = -1
keepdims = 1
select_last_index = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 4, 5, 1), dtype=int64
[[[[0],
[1],
[2],
[4],
[2]],
[[0],
[5],
[2],
[4],
[4]],
[[3],
[0],
[3],
[1],
[3]],
[[3],
[2],
[4],
[3],
[4]]],
[[[5],
[5],
[0],
[1],
[1]],
[[1],
[1],
[4],
[1],
[0]],
[[4],
[3],
[5],
[5],
[5]],
[[0],
[0],
[0],
[5],
[5]]],
[[[5],
[4],
[1],
[4],
[3]],
[[5],
[5],
[1],
[5],
[4]],
[[1],
[0],
[2],
[5],
[3]],
[[4],
[4],
[2],
[5],
[2]]]]
test_cc_argmax_no_keepdims_example
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 0
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2,), dtype=int64
[0, 1]
test_cc_argmax_no_keepdims_example_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 0
select_last_index = 1
Inputs:
data: shape=(2, 2), dtype=float32
[[ 2., 2.],
[ 3., 10.]]
Outputs:
result: shape=(2,), dtype=int64
[1, 1]
test_cc_argmax_no_keepdims_random
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 0
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 5, 6), dtype=int64
[[[0, 0, 2, 2, 3, 2],
[2, 0, 3, 0, 3, 1],
[3, 3, 1, 2, 3, 0],
[2, 2, 2, 3, 0, 0],
[1, 1, 0, 2, 3, 2]],
[[3, 1, 1, 0, 2, 0],
[1, 1, 0, 2, 0, 0],
[3, 2, 0, 3, 1, 2],
[0, 0, 0, 3, 0, 3],
[1, 0, 1, 1, 0, 2]],
[[1, 2, 2, 0, 0, 1],
[2, 2, 0, 0, 3, 0],
[1, 1, 2, 0, 1, 0],
[3, 1, 2, 1, 0, 2],
[0, 3, 3, 0, 1, 1]]]
test_cc_argmax_no_keepdims_random_select_last_index
Node:
ArgMax(data) -> (result)
Attributes:
axis = 1
keepdims = 0
select_last_index = 1
Inputs:
data: shape=(3, 4, 5, 6), dtype=float32
[[[[ 1.1535902 , 0.8930068 , 0.31370395, 0.75162536, -0.8198946 ,
0.5718451 ],
[ 0.48409304, 0.9352933 , -0.3099123 , 0.796408 , -1.0177791 ,
0.19120698],
[-0.86700433, 0.03553583, 1.6169748 , 0.13775356, -0.30418357,
0.92265576],
[-0.24711125, -0.6013053 , -1.7442979 , -0.24002658, 1.1893611 ,
0.3885773 ],
[-0.881166 , -0.26651508, 0.51943284, -0.5207664 , 0.49173778,
-1.4098834 ]],
[[ 0.2084414 , -0.1663104 , -0.5272354 , -0.7789604 , -1.3383248 ,
0.20150597],
[ 0.18964349, -0.9699509 , 0.01368247, 0.26326245, -0.11800206,
1.0865252 ],
[-1.7502077 , 0.4797004 , 2.016871 , 1.0357574 , -0.38723874,
-0.6893609 ],
[-0.08571087, -0.12568739, -0.45894295, -2.4101048 , -0.05234696,
-0.32065794],
[-0.0337468 , 0.44579372, -0.31529766, -0.35173053, 1.4849298 ,
0.7717439 ]],
[[ 0.74211425, 0.41953868, 0.8878202 , 1.9992762 , -1.2793657 ,
1.9040331 ],
[ 1.3186278 , 0.09078357, -1.5114992 , -0.5912303 , -0.10834445,
-2.0142362 ],
[-1.8717729 , -1.5980401 , 1.0740488 , 1.629672 , -0.10766515,
0.06840862],
[ 0.8648494 , 1.5139848 , 1.3529935 , -0.6761605 , -0.24604046,
-1.3292124 ],
[-0.41538328, 0.23129524, -0.7840473 , 1.3079971 , 1.0258558 ,
1.2286781 ]],
[[-0.19445363, 0.612231 , -1.1722137 , 1.5029458 , 0.22012453,
0.02289531],
[-1.5329183 , -1.2377479 , 0.7384988 , -0.4947421 , 0.5249709 ,
-1.1028512 ],
[-0.50560296, 0.64586425, -0.37096298, 1.0453857 , 1.1110809 ,
0.63471276],
[-1.7674397 , 0.05805698, 1.0269951 , 1.2926332 , -0.8670364 ,
-0.04096305],
[-1.3972218 , -0.56706476, -1.5361938 , 0.7568297 , 1.9624395 ,
0.01612921]]],
[[[-0.8381156 , 0.44593495, -0.7598965 , 0.8841587 , -0.4179815 ,
1.2437001 ],
[ 0.6833002 , -1.5051733 , -0.04142308, -1.0440199 , 0.45535547,
0.8325864 ],
[ 2.6598756 , -1.0611902 , 1.3398936 , -0.63711953, 0.8682988 ,
-1.0831335 ],
[ 0.14741649, 2.456468 , 0.20894337, -0.03364305, 0.6039444 ,
-0.4583244 ],
[-1.8623905 , 1.8956882 , -1.0354153 , 1.2715455 , 0.5872212 ,
-0.937322 ]],
[[ 0.3792371 , 0.9923941 , 0.9530322 , -0.19218627, 0.24083266,
0.3095932 ],
[ 0.92882544, 1.1939268 , -0.45487338, -1.0227059 , -0.90328914,
0.43774566],
[-0.16577157, -0.88705194, -0.76002544, 0.23932524, 1.3093162 ,
0.59931517],
[ 0.03689478, 1.3963245 , -0.41871142, -1.2458491 , -1.7682317 ,
-0.8284078 ],
[ 2.6940649 , 0.20736967, 1.4835985 , 2.1304066 , -0.7621148 ,
-0.17221664]],
[[-2.5051978 , -0.86628616, -1.762865 , -0.00515244, 1.3064455 ,
0.30165917],
[-0.50755274, -0.03781877, -1.1887847 , 0.13941556, -0.6613395 ,
-1.3875629 ],
[ 1.218692 , 0.24355745, -0.71431977, -2.237271 , 0.5288873 ,
1.3125248 ],
[ 0.05970533, -1.0772544 , -1.5295768 , -0.06837761, -0.05408159,
0.14961782],
[ 0.5556948 , -2.372416 , 0.17837672, -1.2947078 , 0.1843701 ,
1.7052302 ]],
[[ 1.8640412 , -1.6172435 , 0.11948529, 0.6397907 , 1.1356219 ,
-1.3952476 ],
[ 0.0460656 , -1.4542258 , -1.7263715 , -0.48677343, -0.6715086 ,
-1.154739 ],
[ 3.0360975 , -0.27158165, -0.4962192 , 0.5677865 , 0.41674662,
0.0303928 ],
[-1.2915128 , -0.9910574 , 0.15541704, 0.3814489 , -0.8719388 ,
1.9986113 ],
[ 0.3670998 , -0.2306997 , 0.03010399, 0.09262651, 0.47344238,
1.5063039 ]]],
[[[-0.87134457, 0.49287248, 0.96859217, 0.10066097, 0.20803922,
1.3550882 ],
[-1.3531399 , 0.09047035, 0.72713685, 0.66012996, 1.3432837 ,
0.7859343 ],
[ 0.97664225, 1.3816993 , -0.58079594, 0.5459451 , -0.51179546,
0.6222708 ],
[-0.4023195 , -0.62198836, -0.20893596, -1.6246839 , 0.5678284 ,
-0.24042673],
[ 0.07070588, 0.12409985, -1.8788625 , 1.4642497 , -1.8194745 ,
-1.3071544 ]],
[[ 0.5802228 , -0.30956224, -0.8219037 , -0.03411172, -0.06197588,
1.7122005 ],
[-0.7808519 , -2.2874446 , 0.38231578, -0.4402865 , -2.2392201 ,
0.3882465 ],
[ 1.0887306 , 1.6925359 , 0.480331 , 0.1313244 , 1.1624386 ,
-0.09964492],
[-2.6772964 , 1.2427499 , -0.6564615 , 0.9886259 , 0.43802845,
1.3261851 ],
[-0.59629834, -1.954724 , -0.90917647, 0.27276233, 1.9002748 ,
1.3088909 ]],
[[-1.3319217 , 1.8170048 , 1.1183677 , -0.2203691 , -2.3447175 ,
-0.62422806],
[ 1.2510281 , 0.4823911 , -0.83907837, -1.0868605 , -0.61670125,
0.33403757],
[ 0.72298384, 0.17178722, 0.73810834, -2.5713859 , -0.06512517,
-0.03371745],
[-1.3262166 , 0.02683339, 0.04087868, 0.9181843 , -0.947292 ,
1.484566 ],
[-0.06386357, -0.44807073, 0.4431652 , 1.0376054 , 0.6025506 ,
-0.45821458]],
[[-0.258412 , -1.630167 , -0.5343529 , -0.9920857 , -0.13553081,
-2.3305297 ],
[ 0.3500176 , -0.02892778, -0.23109433, -0.60780907, 1.4199332 ,
-0.94604033],
[-0.6791459 , -0.77185553, 0.02125573, -1.1943868 , -0.45757383,
-1.9794677 ],
[-0.05192753, -1.3392941 , -0.56291044, -1.3561316 , -1.2371935 ,
1.1204599 ],
[-0.3621374 , 0.2506451 , 2.741324 , 1.1915846 , -0.92346644,
-0.925773 ]]]]
Outputs:
result: shape=(3, 5, 6), dtype=int64
[[[0, 0, 2, 2, 3, 2],
[2, 0, 3, 0, 3, 1],
[3, 3, 1, 2, 3, 0],
[2, 2, 2, 3, 0, 0],
[1, 1, 0, 2, 3, 2]],
[[3, 1, 1, 0, 2, 0],
[1, 1, 0, 2, 0, 0],
[3, 2, 0, 3, 1, 2],
[0, 0, 0, 3, 0, 3],
[1, 0, 1, 1, 0, 2]],
[[1, 2, 2, 0, 0, 1],
[2, 2, 0, 0, 3, 0],
[1, 1, 2, 0, 1, 0],
[3, 1, 2, 1, 0, 2],
[0, 3, 3, 0, 1, 1]]]
Differences with previous version (12)#
SchemaDiff: ArgMax (domain 'ai.onnx')
old version: 12
new version: 13
breaking: no
Type constraints:
changed ‘T’: added types: [‘tensor(bfloat16)’]