TensorLabel

TensorLabel は、軸上に意味のあるラベルを持つ TensorBuffers のユーティリティ ラッパーです。

たとえば、画像分類モデルは、{1, 10} のような形状を持つ出力テンソルを持つ場合があります。ここで、1 はバッチ サイズ、10 はカテゴリの数です。実際、2 番目の軸では、対応する各カテゴリの名前または説明を使用して各サブテンソルにラベルを付けることができます。 TensorLabel TensorBuffer内のプレーン Tensor を、事前定義されたラベルからサブテンソルへのマップに変換するのに役立ちます。この場合、2 番目の軸に 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 より大きい最初のラベルのテンソルからマップへの変換のみをサポートしています。