将 ND4J INDArray 转换为列表

问题描述 投票:0回答:1

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

我可以多次转换对象,使用

myIndarray.data()
来获取 DataBuffer,然后转换 DataBuffer 来获取 int 数组,然后获取 ArrayList,但这似乎有点牵强:

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

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

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

java nd4j
1个回答
0
投票

也许我找到了答案:NDArrayUtil.toInts,采用

INDArray
并输出
int[]
,但没有相关文档。

© www.soinside.com 2019 - 2024. All rights reserved.