TensorLabel

TensorLabel হল একটি অক্ষের উপর অর্থপূর্ণ লেবেল সহ TensorBuffers-এর জন্য একটি util wrapper।

উদাহরণস্বরূপ, একটি ইমেজ শ্রেণীবিভাগ মডেলের একটি আউটপুট টেনসর থাকতে পারে যার আকৃতি {1, 10}, যেখানে 1 হল ব্যাচের আকার এবং 10 হল বিভাগের সংখ্যা। প্রকৃতপক্ষে, ২য় অক্ষে, আমরা প্রতিটি সাব-টেনসরকে প্রতিটি সংশ্লিষ্ট বিভাগের নাম বা বিবরণ দিয়ে লেবেল করতে পারি। TensorLabel TensorBuffer প্লেইন টেনসরকে পূর্বনির্ধারিত লেবেল থেকে সাব-টেনসরে রূপান্তর করতে সাহায্য করতে পারে। এই ক্ষেত্রে, যদি দ্বিতীয় অক্ষের জন্য 10টি লেবেল প্রদান করা হয়, TensorLabel মূল {1, 10} টেনসরকে একটি 10 ​​উপাদান মানচিত্রে রূপান্তর করতে পারে, যার প্রতিটি মান টেনসর আকারে {} (স্কেলার)। ব্যবহারের উদাহরণ:

   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 এর বেশি আকারের প্রথম লেবেলের জন্য শুধুমাত্র টেনসর-টু-ম্যাপ রূপান্তর সমর্থন করি।