| { |
| "tests": [ |
| { |
| "name": "argMin float32 1D tensor default options", |
| "inputs": { |
| "input": { |
| "shape": [24], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 2D tensor default options", |
| "inputs": { |
| "input": { |
| "shape": [4, 6], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 3D tensor default options", |
| "inputs": { |
| "input": { |
| "shape": [2, 3, 4], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor default options", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 5D tensor default options", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 1, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[2]", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [2] |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [2, 1, 3], |
| "data": [ |
| 2, |
| 2, |
| 0, |
| 0, |
| 0, |
| 0 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[]", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [] |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0, |
| 0 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.keepDimensions=true", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "keepDimensions": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 1, 1, 1], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.keepDimensions=false", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "keepDimensions": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.selectLastIndex=true", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "selectLastIndex": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 19 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.selectLastIndex=false", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "selectLastIndex": false |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 7 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[0, 2] options.keepDimensions=false", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [0, 2], |
| "keepDimensions": false |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 3], |
| "data": [ |
| 2, |
| 2, |
| 4 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.keepDimensions=true", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [3, 0, 1], |
| "keepDimensions": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 1, 4, 1], |
| "data": [ |
| 4, |
| 0, |
| 1, |
| 1 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[0, 2] options.selectLastIndex=false", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [0, 2], |
| "selectLastIndex": false |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 3], |
| "data": [ |
| 2, |
| 2, |
| 4 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[0, 2] options.selectLastIndex=true", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [0, 2], |
| "selectLastIndex": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 3], |
| "data": [ |
| 6, |
| 6, |
| 4 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.selectLastIndex=false", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [3, 0, 1], |
| "selectLastIndex": false |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [4], |
| "data": [ |
| 4, |
| 0, |
| 1, |
| 1 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.selectLastIndex=true", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [3, 0, 1], |
| "selectLastIndex": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [4], |
| "data": [ |
| 4, |
| 0, |
| 4, |
| 1 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 4D tensor all options", |
| "inputs": { |
| "input": { |
| "shape": [2, 1, 4, 3], |
| "data": [ |
| 3.830124090690262, |
| -24.986487937638074, |
| 5.299982630691289, |
| -48.5486590218902, |
| 40.30886781808215, |
| 60.184293919409726, |
| -82.78385618759043, |
| -96.50904103637833, |
| 71.87028201591897, |
| 38.866394268784035, |
| -39.143725517854435, |
| 31.444366685561903, |
| -82.78385618759043, |
| -96.50904103637833, |
| -25.53388886326502, |
| -16.142265850469343, |
| 66.63677406472371, |
| 82.5119815304117, |
| -82.78385618759043, |
| -96.50904103637833, |
| 39.7687246127592, |
| 42.15040238450999, |
| 82.66863662444459, |
| 85.4526923278379 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [3, 0], |
| "keepDimensions": true, |
| "selectLastIndex": false |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [1, 1, 4, 1], |
| "data": [ |
| 4, |
| 0, |
| 1, |
| 1 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 0D scalar options.axes=[]", |
| "inputs": { |
| "input": { |
| "shape": [], |
| "data": [ |
| 3.830124090690262 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [] |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 0 |
| ], |
| "type": "int64" |
| } |
| }, |
| { |
| "name": "argMin float32 0D scalar options.axes=[] no effect by both keepDimensions and selectLastIndex being true", |
| "inputs": { |
| "input": { |
| "shape": [], |
| "data": [ |
| 3.830124090690262 |
| ], |
| "type": "float32" |
| } |
| }, |
| "options": { |
| "axes": [], |
| "keepDimensions": true, |
| "selectLastIndex": true |
| }, |
| "expected": { |
| "name": "output", |
| "shape": [], |
| "data": [ |
| 0 |
| ], |
| "type": "int64" |
| } |
| } |
| ] |
| } |