TensorLabel

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

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