fix casting of onnxTensor.getValue from String[] to String[][]
This commit is contained in:
Ramin 2023-11-15 05:15:56 +03:30 коммит произвёл GitHub
Родитель 5aeed62630
Коммит 9e06cefd1d
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -113,10 +113,10 @@ public class TensorHelper {
// data
if (onnxTensor.getInfo().onnxType == TensorInfo.OnnxTensorType.ONNX_TENSOR_ELEMENT_DATA_TYPE_STRING) {
String[] buffer = (String[])onnxTensor.getValue();
String[][] buffer = (String[][])onnxTensor.getValue();
WritableArray dataArray = Arguments.createArray();
for (String value : buffer) {
dataArray.pushString(value);
for (String[] value : buffer) {
dataArray.pushString(value[0]);
}
outputTensor.putArray("data", dataArray);
} else {