TensorLabel

TensorLabel은 축에 의미 있는 라벨이 있는 TensorBuffers용 유틸리티 래퍼입니다.

예를 들어 이미지 분류 모델에 형태가 {1, 10}인 출력 텐서가 있을 수 있습니다. 여기서 1은 배치 크기이고 10은 카테고리 수입니다. 실제로 두 번째 축에서 각 하위 텐서에 해당하는 각 카테고리의 이름 또는 설명으로 라벨을 지정할 수 있습니다. TensorLabel는 TensorBuffer의 일반 텐서를 사전 정의된 라벨에서 하위 텐서로의 매핑으로 변환하는 데 도움이 될 수 있습니다. 이 경우, 두 번째 축에 라벨 10개를 제공하면 TensorLabel가 원래 {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();
 

참고: 현재 크기가 1보다 큰 첫 번째 라벨에만 텐서-맵 변환이 지원됩니다.