nd4j 数组返回不正确的排名

Question

当我从 1d float[] 创建一个 nd 数组时，它的排名为 2。

INDArray array = Nd4j.create(new float[] {0,1,2,3,4,5});
int rank = array.rank(); // Is 2!

根据文档，这应该是排名 1：http://nd4j.org/userguide#createfromjava

任何想法如何创建等级 1 INDArray？

Answer 1

我不知道为什么这很重要。您不需要 1 秩数组。我们故意这样构建 nd4j。这避免了各种边缘情况。

形状为 1 x 1 的 2 秩数组仍然是长度为 1 的数组。