{ "tests": [ { "name": "argMin float32 1D constant 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", "constant": true } }, "expected": { "name": "output", "shape": [], "data": [ 7 ], "type": "int64" } }, { "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" } } ] }