如何将Json中的n维矩阵转换为java中的INDArray

最新推荐文章于 2024-02-01 00:49:25 发布

Wendell6

最新推荐文章于 2024-02-01 00:49:25 发布

阅读量881

点赞数

分类专栏： json Nd4j c++ 机器学习文章标签： json marix nd4j

本文链接：https://blog.csdn.net/wwwwenming2007/article/details/54630019

版权

c++ 同时被 3 个专栏收录

4 篇文章 0 订阅

订阅专栏

机器学习

4 篇文章 0 订阅

订阅专栏

Nd4j

3 篇文章 0 订阅

订阅专栏

代码使用了递归， gson，nd4j

import com.google.gson.JsonElement;
import org.apache.commons.lang3.ArrayUtils;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.api.shape.Shape;

import java.util.ArrayList;
import java.util.Arrays;

class CommonPort{
    static INDArray load_json_to_NDArray(JsonElement json_array, int depth) {
        if (0 >= depth)
            return null;

        ArrayList<Integer> shape = get_shape(json_array, depth);
        Integer[] shape_a = new Integer[shape.size()];
        shape.toArray(shape_a);
        int[] shape_int = ArrayUtils.toPrimitive(shape_a);

        INDArray result;
        if (1 == shape_int.length)
            result = Nd4j.create(shape_int[0]);
        else
            result = Nd4j.zeros(shape_int);

        int json_array_size = json_array.getAsJsonArray().size();
        for(int i = 0; i < json_array_size; ++i){
            if (1 == depth)
                result.putScalar(i, json_array.getAsJsonArray().get(i).getAsDouble());
            else
                result.putRow(i, load_json_to_NDArray(json_array.getAsJsonArray().get(i), depth - 1));
        }

        if (1 == shape_int.length){
            result = result.permute(1, 0);
        }

        return result;
    }

    static private ArrayList<Integer> get_shape(JsonElement json_array, int depth) {
        ArrayList<Integer> result = new ArrayList<>();

        if (1 == depth)
            result.add(json_array.getAsJsonArray().size());
        else if (1 < depth){
            result.add(json_array.getAsJsonArray().size());
            result.addAll(get_shape(json_array.getAsJsonArray().get(0), depth - 1));
        }
        return result;
    }

}