gpt4 book ai didi

java - 将 ND4J INDArray 转换为列表

转载 作者:行者123 更新时间:2023-11-30 06:10:00 27 4
gpt4 key购买 nike

我正在搜索如何将 N 行和 1 列的 INDArray 转换为 List(ArrayList 或其他东西,无关紧要)对象。

我可以使用 myIndarray.data() 多次转换该对象获取一个DataBuffer,然后将DataBuffer转换为一个int数组,然后是一个ArrayList,但这似乎有点牵强:

List myList = Arrays.asList(myArray.data().asInt());

另一种解决方案是迭代数组并将所有元素放入 ArrayList 中,但这似乎效率不高。

有没有一种干净的方法可以做到这一点?

最佳答案

也许我找到了答案:NDArrayUtil.toInts ,采用 INDArray 并输出 int[],但没有关联的文档。

关于java - 将 ND4J INDArray 转换为列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50468733/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com