nd4j数组返回错误的排名

时间:2017-10-19 16:08:13

标签: java dl4j nd4j

当我从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?

1 个答案:

答案 0 :(得分:0)

我不确定为什么这是一个大问题。你不需要拥有1级数组。我们故意这样建造了nd4j。这避免了各种边缘情况。

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