TensorLabel

TensorLabel הוא מעטפת שימוש עבור TensorBuffers עם תוויות משמעותיות על ציר.

לדוגמה, למודל סיווג תמונה עשוי להיות טנזור פלט עם צורה כמו {1, 10}, כאשר 1 הוא גודל האצווה ו-10 הוא מספר הקטגוריות. למעשה, בציר השני, נוכל לתייג כל תת-טנזור עם השם או התיאור של כל קטגוריה מתאימה. TensorLabel יכול לעזור בהמרת Tensor הפשוט ב- TensorBuffer למפה מתוויות מוגדרות מראש לתת-טנסורים. במקרה זה, אם יסופקו 10 תוויות עבור הציר השני, TensorLabel יכול להמיר את הטנסור המקורי {1, 10} למפה של 10 אלמנטים, שכל ערך שלה הוא Tensor בצורה {} (סקלרית). דוגמה לשימוש:

   TensorBuffer outputTensor = ...;
   List<String> labels = FileUtil.loadLabels(context, labelFilePath);
   // labels the first axis with size greater than one
   TensorLabel labeled = new TensorLabel(labels, outputTensor);
   // If each sub-tensor has effectively size 1, we can directly get a float value
   Map<String, Float> probabilities = labeled.getMapWithFloatValue();
   // Or get sub-tensors, when each sub-tensor has elements more than 1
   Map<String, TensorBuffer> subTensors = labeled.getMapWithTensorBuffer();
 

הערה: נכון לעכשיו, אנו תומכים רק בהמרת טנזור למפה עבור התווית הראשונה עם גודל גדול מ-1.