TensorLabel

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

לדוגמה, למודל של סיווג תמונות יכול להיות t n sor פלט עם צורה כ-{1, 10}, כאשר 1 הוא גודל האצווה ו-10 הוא מספר הקטגוריות. למעשה, בציר השני יכולנו להוסיף לכל תת-רכיב עם השם או התיאור של כל קטגוריה תואמת. בעזרת TensorLabel אפשר להפוך את Tensor פשוט ב-TensorBuffer למפה מתוויות מוגדרות מראש לערכי משנה. במקרה כזה, אם מספקים 10 תוויות לציר השני, TensorLabel יכול להמיר את Tensor {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();
 

הערה: בשלב זה אנחנו תומכים רק בהמרה מסוג tensor-למפה לתווית הראשונה שהגודל שלה גדול מ-1.