Question

我无法将double转换为Nd4j.INDArray而不会丢失精度。

double[] weight = new double[]{-0.13404223866376802,-0.11294084872465669,0.11232944517596409,-0.01687720880184701,0.13288394029423414,-0.06916641552096867,0.1518882606786481};
INDArray array = Nd4j.create(weight, new int[]{1, 7});
System.out.println(array);

输出

[-0.13,  -0.11,  0.11,  -0.02,  0.13,  -0.07,  0.15]

所需的输出应为

[-0.13404223866376802,-0.11294084872465669,0.11232944517596409,-0.01687720880184701,0.13288394029423414,-0.06916641552096867,0.1518882606786481]

如何以全精度转换？

Answer 1

使用

int precision = 7;
String printed = new NDArrayStrings(precision).format(this);
System.out.println(printed);

你想要的任何精度。 NDArrayStrings位于org.nd4j.linalg.string包中。

Answer 2

我有同样的问题，最新的0.9.x nd4j：

    new NDArrayStrings(15).format(Nd4j.create(new double[]{0.4,0.3}))
    [0.400000005960464,  0.300000011920929]

    Nd4j.create(new double[]{0.4}).getDouble(0)
    0.4000000059604645

编辑：现在我明白了原因，Nd4j仅初始化为“ float”数字... 因此只有7位数字是正确的，最好使用“ getFloat”

    Nd4j.create(new double[]{0.4 }).getFloat(0)
    0.4

Answer 3

在创建INDArray之前使用此行：

Nd4j.setDefaultDataTypes(org.nd4j.linalg.api.buffer.DataType.DOUBLE, org.nd4j.linalg.api.buffer.DataType.FLOAT16);

这会将默认精度Nd4j更改为Double和Float16。

您可以在类中进行静态调用，因此只需调用一次就可以忽略它。

    class Foo {
       static {
          putTheCallHere();
       }
    }

INDArray的Java double损失了精度

3 个答案: