aprender-core 0.65.2

Next-generation machine learning library in pure Rust
{
  "cases": [
    {
      "case_id": "mixed_length_pair",
      "texts": [
        "Short text.",
        "This sentence is deliberately longer so the batch holds two different lengths and padding is exercised."
      ],
      "input_ids_canonical": [
        [101, 2460, 3793, 1012, 102, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [
          101, 2023, 6251, 2003, 9969, 2936, 2061, 1996, 14108, 4324, 2048, 2367, 10742, 1998,
          11687, 4667, 2003, 17747, 1012, 102
        ]
      ],
      "attention_mask": [
        [1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
      ],
      "shape": {
        "batch": 2,
        "hidden": 64
      },
      "pooled": [
        -0.148869306, -0.143380389, -0.0187003762, -0.134526461, 0.130802959, 0.0352705717,
        0.0780760944, 0.197420388, 0.0239898209, -0.254129589, 0.539312422, 0.817723095,
        -0.0763259754, -0.149328128, -0.228271961, 0.0744099468, 0.0794911012, 0.188638374,
        0.0215837471, 0.0295029692, 0.062215209, -0.0026320219, -0.103019454, -0.246660277,
        -0.0780025125, 0.0343635418, -0.259467512, -0.405351967, 0.292984605, 0.0768261924,
        0.104188941, 0.907271385, 0.380234897, 0.0376874879, 0.670534432, 0.209163591,
        -0.0626308322, -0.142521665, 0.152307704, -0.0424722545, -0.234842688, -0.0808362961,
        0.315453112, -0.0577549413, -0.0142836999, -0.337540686, 0.271295846, -0.265866876,
        -0.275187403, -0.172712952, -0.576153636, 0.0561855063, 0.155802324, 0.272260219,
        0.0672519132, -0.390273035, -0.104684755, -0.20807448, 0.000292562705, 0.0800272226,
        -0.243338436, -0.188814119, -0.174659312, 0.966198742, -0.24511829, -0.216296941,
        -0.0159148816, -0.135032386, -0.093990624, 0.120671466, 0.0356783159, -0.235660881,
        -0.0481933914, -0.110694088, 0.478275955, 0.166645259, -0.0194707774, 0.0385940783,
        -0.125192001, -0.155061722, -0.218228295, 0.125900939, -0.138362199, -0.0270730741,
        0.51023978, 0.0717832148, -0.137783453, 0.107347213, -0.0183979515, -0.0328591578,
        -0.0938118845, -0.357941359, 0.486790895, -0.046845641, -0.0263501313, 0.609982133,
        0.0733655095, 0.111460969, 0.381258428, -0.146264061, 0.0635138974, -0.354789436,
        -0.296183765, -0.0981404036, -0.185660437, 0.179000035, -0.0178923756, -0.090796493,
        0.213307664, 0.12808609, 0.155667618, -0.187493578, -0.168372661, -0.139083296,
        -0.0711447597, -0.00785422511, 0.174639076, 0.172881812, -0.0710122064, 0.0253121015,
        0.0799111277, -0.0462135486, 0.150367618, -0.0602268353, 0.315055609, 0.0323234424,
        0.13079825, 0.732548118
      ],
      "normalized": [
        -0.0637961999, -0.0614439882, -0.00801382773, -0.0576497391, 0.056054078, 0.0151147908,
        0.033458598, 0.0846021995, 0.010280557, -0.108904257, 0.231116027, 0.350425661,
        -0.0327086039, -0.0639928207, -0.0978232771, 0.0318875127, 0.0340649821, 0.0808387697,
        0.00924946181, 0.0126431519, 0.0266615991, -0.00112792221, -0.044147782, -0.105703376,
        -0.0334270634, 0.014726094, -0.111191764, -0.173708841, 0.125555128, 0.0329229645,
        0.0446489528, 0.388800532, 0.162945211, 0.0161505323, 0.287349671, 0.0896346122,
        -0.0268397108, -0.0610759929, 0.0652696863, -0.0182009879, -0.100639082, -0.0346414484,
        0.135183737, -0.0247502048, -0.0061211125, -0.144649103, 0.116260663, -0.113934137,
        -0.117928341, -0.0740141124, -0.246903896, 0.0240776408, 0.0667672604, 0.116673931,
        0.0288200211, -0.167246938, -0.0448614284, -0.0891678855, 0.000125374325, 0.0342947282,
        -0.104279839, -0.0809140801, -0.0748482049, 0.414053172, -0.140039116, -0.123573117,
        -0.00909236819, -0.0771456733, -0.053698007, 0.0689411014, 0.0203834623, -0.13463597,
        -0.0275334809, -0.0632409006, 0.273244977, 0.0952064991, -0.0111238956, 0.0220492743,
        -0.0715237409, -0.0885886773, -0.124676526, 0.0719287619, -0.0790480375, -0.0154671818,
        0.29150629, 0.0410106368, -0.0787173882, 0.0613287911, -0.0105109774, -0.0187728424,
        -0.0535958894, -0.204496324, 0.27810967, -0.0267634932, -0.0150541561, 0.348490328,
        0.0419146232, 0.063679032, 0.217817649, -0.0835624635, 0.0362862758, -0.202695593,
        -0.169213444, -0.0560688265, -0.106070101, 0.102264933, -0.0102221351, -0.0518731587,
        0.121865302, 0.0731771663, 0.0889348313, -0.1071174, -0.0961933807, -0.0794600099,
        -0.0406458825, -0.00448721601, 0.0997734666, 0.0987695158, -0.0405701511, 0.0144611169,
        0.0456542149, -0.026402371, 0.0859068781, -0.0344083346, 0.179995164, 0.0184667818,
        0.0747266561, 0.418513805
      ]
    },
    {
      "case_id": "pooling_batch",
      "texts": [
        "One line.",
        "Another line that is a good deal longer than the first one.",
        "Third."
      ],
      "input_ids_canonical": [
        [101, 2028, 2240, 1012, 102, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [
          101, 2178, 2240, 2008, 2003, 1037, 2204, 3066, 2936, 2084, 1996, 2034, 2028, 1012,
          102
        ],
        [101, 2353, 1012, 102, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
      ],
      "attention_mask": [
        [1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
        [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
      ],
      "shape": {
        "batch": 3,
        "hidden": 64
      },
      "pooled": [
        -0.632604897, -0.302709192, 0.232466221, -0.0363686606, 0.0239333455, 0.340752602,
        0.469153315, -0.108144283, -0.226031691, -0.0332050435, 0.475518376, 0.674385548,
        0.00725292647, -0.0181683414, -0.146145731, 0.000959432102, 0.283959806, -0.108088061,
        0.29386729, 0.184583813, -0.681637108, 0.37263003, -0.28056711, -0.107058488,
        -0.109244466, 0.0237193797, -0.150277808, -0.356545985, -0.657320678, -0.0669947267,
        -0.0557126291, 0.752761602, -0.030682575, 0.139573395, 0.600702822, 0.29455021,
        -0.156947121, -0.114490949, -0.119168058, -0.0505924448, -0.171972156, 0.0811228976,
        -0.0306651294, 0.0805189833, 0.0995755196, -0.121709622, 0.157171503, -0.272361249,
        -0.318453819, -0.0517904274, -0.143793419, 0.0744674355, 0.186557904, -0.0591908395,
        -0.104821786, 0.0104449689, -0.172699288, -0.0823858529, 0.100458264, 0.110509411,
        -0.187355861, 0.015315339, 0.282345146, 1.06984937, -0.583439887, 0.0661133379,
        0.121572211, 0.099734664, 0.0459128283, 0.377507508, -0.0686322078, 0.0605868436,
        -0.0377707966, -0.0963306576, 0.390599936, 0.412826896, -0.0688043013, 0.0935029835,
        0.0718725398, -0.0758854225, 0.108339444, -0.0345673077, 0.243337885, -0.238202095,
        -0.38879931, 0.218998656, -0.0201321337, -0.00118290831, -0.170749336, 0.00419699214,
        -0.115521766, -0.284239143, 0.0254705455, -0.116486788, -0.154150903, 0.339051247,
        0.205376804, 0.040137466, 0.428895593, -0.0970205292, -0.0162759442, -0.17616871,
        -0.236814305, -0.166335166, -0.309609592, 0.0418079495, 0.0514432676, -0.0507159121,
        0.215307847, -0.199310362, 0.316457242, -0.347247481, -0.15679656, -0.0866521224,
        0.171358019, -0.0557984821, 0.278349012, -0.038668599, -0.0365000926, 0.0174162667,
        0.0123912888, -0.351126075, 0.11952085, -0.142889202, 0.0707818046, 0.0381683856,
        0.373283446, 1.08448672, -0.363146424, 0.115369543, 0.0935785621, -0.0844237134,
        0.43869108, 0.42597729, 0.00509160757, -0.0722870827, -0.311766744, 0.0993092284,
        0.807328582, 0.553079069, -0.110202335, -0.0776906833, -0.347798616, 0.184649765,
        0.448711962, -0.153052911, 0.0909389481, 0.325723469, -0.747887731, 0.0393312275,
        0.0244960785, -0.103594147, 0.156397104, 0.161283895, -0.177476764, -0.2508834,
        -0.62796551, 0.014547158, 0.0121069672, 0.461840123, 0.00812499225, -0.0364379138,
        0.612594485, 0.277385384, 0.0887674987, -0.276312083, 0.0817276537, -0.214188322,
        -0.102112591, 0.0983513817, 0.270008773, -0.0832000673, 0.214428872, -0.303181201,
        0.266118199, -0.342745841, -0.439913481, -0.27105993, -0.671311617, 0.000155286863,
        0.0977476686, 0.068268612, -0.171870559, -0.316638649, -0.164112583, -0.210723311,
        0.202375501, 0.243565306, -0.0315839946, -0.1243136, 0.0247244425, 1.17166293
      ],
      "normalized": [
        -0.26014334, -0.124481775, 0.0955960676, -0.0149557255, 0.00984200463, 0.140126199,
        0.192927867, -0.0444716997, -0.0929500237, -0.0136547647, 0.195545346, 0.277324617,
        0.00298258918, -0.00747128762, -0.0600988716, 0.00039454308, 0.116771549,
        -0.0444485806, 0.120845757, 0.0759055912, -0.280306637, 0.153235018, -0.115376383,
        -0.0440251939, -0.0449241251, 0.009754017, -0.0617980845, -0.14662084, -0.270307094,
        -0.0275499485, -0.0229104608, 0.309554875, -0.0126174614, 0.0573961586, 0.247024387,
        0.121126592, -0.0645406768, -0.0470816121, -0.0490049608, -0.02080491, -0.0707193539,
        0.0333598144, -0.0126102874, 0.033111468, 0.0409480073, -0.0500501171, 0.0646329522,
        -0.112001926, -0.130956367, -0.0212975517, -0.0591315404, 0.0306229182, 0.0767173916,
        -0.0243407898, -0.0431054048, 0.00429523876, -0.0710183755, -0.0338791758,
        0.0413110144, 0.0454443023, -0.0770455301, 0.00629806006, 0.116107561, 0.439949483,
        -0.296489269, 0.0335971117, 0.0617798977, 0.0506826118, 0.0233317278, 0.19183968,
        -0.0348771363, 0.030788688, -0.0191941559, -0.0489527807, 0.198492914, 0.209788099,
        -0.0349645913, 0.0475158319, 0.0365237892, -0.0385630354, 0.0550553426, -0.0175662246,
        0.123658106, -0.121048227, -0.197577894, 0.111289531, -0.0102306372, -0.000601123844,
        -0.0867704526, 0.00213280437, -0.0587052144, -0.14444308, 0.0129434811, -0.0591956154,
        -0.0783355534, 0.172297195, 0.104367249, 0.0203968361, 0.217953801, -0.0493033566,
        -0.00827101991, -0.0895244405, -0.120342985, -0.0845272839, -0.157335699, 0.0212457329,
        0.0261421558, -0.0257725324, 0.109413952, -0.101284444, 0.160815492, -0.176462308,
        -0.0796800107, -0.0440343991, 0.0870797709, -0.0283553656, 0.141449869, -0.019650396,
        -0.018548416, 0.00885050278, 0.00629693689, -0.178433314, 0.060737446, -0.0726126432,
        0.0359695069, 0.0193961989, 0.189693123, 0.551108479, -0.138426974, 0.043977458,
        0.0356710032, -0.032181289, 0.167223677, 0.162377328, 0.00194085855, -0.0275549516,
        -0.11884167, 0.037855465, 0.307743788, 0.210826978, -0.0420077816, -0.0296147391,
        -0.132576585, 0.0703862309, 0.171043515, -0.058341898, 0.0346648134, 0.124161802,
        -0.285085648, 0.0149925835, 0.00933760591, -0.0394888222, 0.0596166626, 0.0614794493,
        -0.0676519722, -0.0956336856, -0.239372775, 0.00554519892, 0.00461502792, 0.176047802,
        0.00309714768, -0.0138896871, 0.233513534, 0.10573592, 0.0338370837, -0.105326787,
        0.031153582, -0.081645973, -0.0389240719, 0.0374903455, 0.102924041, -0.0317148492,
        0.0817376673, -0.115568966, 0.101441003, -0.13065052, -0.167689636, -0.103324734,
        -0.255895793, 5.91934549e-05, 0.0372602157, 0.0260231607, -0.0655149594, -0.12069878,
        -0.0625577122, -0.080325149, 0.0771430656, 0.0928441137, -0.0120394323, -0.0473868251,
        0.00942465477, 0.446623594
      ]
    }
  ]
}