Source code
Revision control
Copy as Markdown
Other Tools
{
"tests": [
{
"name": "softmax float32 2D constant tensor all positive",
"inputs": {
"x": {
"shape": [4, 6],
"data": [
7.903734730203054,
6.358251623091922,
4.833756774572475,
9.579165187926348,
0.21071856783621445,
4.554958961778823,
7.150174113718282,
8.330297576341172,
1.535985791562966,
6.633619497318013,
1.45393689965033,
0.21341864769940821,
5.257819475802503,
8.1921378632378,
8.161730206898543,
2.8744343334015565,
8.950733525100087,
6.111632812590415,
1.6371468489476149,
0.2762612074523485,
5.028227187056009,
3.898326029997037,
2.8967114227178836,
6.8894703393234575
],
"type": "float32",
"constant": true
}
},
"expected": {
"name": "output",
"shape": [4, 6],
"data": [
0.15068615972995758,
0.03212761878967285,
0.006995180621743202,
0.8048291206359863,
0.00006871300138300285,
0.005293202120810747,
0.2057899534702301,
0.6698001027107239,
0.0007502624066546559,
0.1227685883641243,
0.0006911618984304368,
0.00019990770670119673,
0.012398251332342625,
0.23319464921951294,
0.22621041536331177,
0.0011435872875154018,
0.4979347288608551,
0.029118351638317108,
0.004253828432410955,
0.001090824487619102,
0.12633030116558075,
0.040812913328409195,
0.014990009367465973,
0.8125221133232117
],
"type": "float32"
}
},
{
"name": "softmax float32 2D tensor all positive",
"inputs": {
"x": {
"shape": [4, 6],
"data": [
7.903734730203054,
6.358251623091922,
4.833756774572475,
9.579165187926348,
0.21071856783621445,
4.554958961778823,
7.150174113718282,
8.330297576341172,
1.535985791562966,
6.633619497318013,
1.45393689965033,
0.21341864769940821,
5.257819475802503,
8.1921378632378,
8.161730206898543,
2.8744343334015565,
8.950733525100087,
6.111632812590415,
1.6371468489476149,
0.2762612074523485,
5.028227187056009,
3.898326029997037,
2.8967114227178836,
6.8894703393234575
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [4, 6],
"data": [
0.15068615972995758,
0.03212761878967285,
0.006995180621743202,
0.8048291206359863,
0.00006871300138300285,
0.005293202120810747,
0.2057899534702301,
0.6698001027107239,
0.0007502624066546559,
0.1227685883641243,
0.0006911618984304368,
0.00019990770670119673,
0.012398251332342625,
0.23319464921951294,
0.22621041536331177,
0.0011435872875154018,
0.4979347288608551,
0.029118351638317108,
0.004253828432410955,
0.001090824487619102,
0.12633030116558075,
0.040812913328409195,
0.014990009367465973,
0.8125221133232117
],
"type": "float32"
}
},
{
"name": "softmax float32 2D tensor all negative",
"inputs": {
"x": {
"shape": [4, 6],
"data": [
-3.3118434934513763,
-3.338954812269648,
-3.4102789638525355,
-6.697192980301335,
-7.896223320202411,
-3.3081689006635866,
-3.2309720485303473,
-4.315771351061817,
-9.311088112018716,
-3.9236627720798856,
-3.7807213277512064,
-6.034926915403203,
-3.9196677453714273,
-2.223484249593408,
-9.326531526135806,
-1.4882492282868132,
-6.302842393071306,
-5.531475594479687,
-1.8421411696525603,
-4.994808274460977,
-9.527291818875955,
-4.985682906583053,
-8.421041200931548,
-6.235629579015209
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [4, 6],
"data": [
0.2546302080154419,
0.24781952798366547,
0.2307596504688263,
0.008623254485428333,
0.002599793951958418,
0.2555675804615021,
0.40352678298950195,
0.13637976348400116,
0.0009232329903170466,
0.20185552537441254,
0.23287305235862732,
0.024441635236144066,
0.0551743283867836,
0.3008708655834198,
0.0002474947541486472,
0.6276082992553711,
0.0050902292132377625,
0.011008745059370995,
0.9090295433998108,
0.0388500951230526,
0.00041779119055718184,
0.039206232875585556,
0.0012629841221496463,
0.011233373545110226
],
"type": "float32"
}
}
]
}