TensorLabel

TensorLabel เป็น wrapper util สำหรับ TensorBuffers ที่มีป้ายกำกับที่มีความหมายบนแกน

ตัวอย่างเช่น โมเดลการจัดหมวดหมู่รูปภาพอาจมีเอาท์พุตเทนเซอร์ที่มีรูปร่างเป็น {1, 10} โดยที่ 1 คือขนาดแบตช์ และ 10 คือจำนวนหมวดหมู่ ที่จริงแล้ว บนแกนที่ 2 เราสามารถติดป้ายกำกับแต่ละเทนเซอร์ย่อยด้วยชื่อหรือคำอธิบายของแต่ละหมวดหมู่ที่เกี่ยวข้องได้ TensorLabel สามารถช่วยแปลง Tensor ธรรมดาใน TensorBuffer ให้เป็นแผนที่จากป้ายกำกับที่กำหนดไว้ล่วงหน้าไปเป็นเทนเซอร์ย่อย ในกรณีนี้ หากระบุป้ายกำกับ 10 รายการสำหรับแกนที่ 2 TensorLabel จะสามารถแปลง {1, 10} Tensor ดั้งเดิมเป็นแผนผังองค์ประกอบ 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