java - nd4j 数组返回不正确的排名

Question

当我从 1d 创建一个 nd 数组时，float[]它的排名为 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 的数组。