【问题标题】:Java double to INDArray loses precisionJava double to INDArray 失去精度
【发布时间】:2018-05-09 04:25:58
【问题描述】:

我无法在不丢失精度的情况下将 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]

如何进行全精度转换?

【问题讨论】:

  • @LeoAso 所说的背后的重点是 INDArray 并没有实际上截断您的数据——它存储和操作整个双精度数。这正是它的 toString() 方法所做的。他的回答允许您以所需的精度格式化 INDArray。
  • @TheAIArchitect 这很有趣。我刚刚使用了 INDArray 的 toDoubleMatrix 并检查了这个。看来我得到的结果与用于创建 INDArray 的原始 double[][] 不同,从小数点后第 7 位开始。看起来数据被截断为FloatBuffer,当转换回Double[][]时,随机垃圾被添加到末尾。

标签: java precision nd4j


【解决方案1】:

在创建 INDArray 之前使用这一行:

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

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

你可以在你的类中放置一个静态调用,这样你调用一次就忘记了。

    class Foo {
       static {
          putTheCallHere();
       }
    }

【讨论】:

    【解决方案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 只初始化为“浮动”数字...... 所以只有7位是正确的,最好使用“getFloat”

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

    【讨论】:

      【解决方案3】:

      使用

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

      无论你想要什么精度。 NDArrayStringsorg.nd4j.linalg.string 包中。

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 2019-10-23
        • 1970-01-01
        • 1970-01-01
        • 2013-07-21
        • 1970-01-01
        • 2012-05-14
        • 1970-01-01
        • 2014-01-13
        相关资源
        最近更新 更多