4

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

如何全精度转换？

4

3 回答 3

1

利用

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

以任何你想要的精度。NDArrayStrings在org.nd4j.linalg.string包装中。

于 2017-11-25T12:02:34.560 回答
0

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

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

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

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

    class Foo {
       static {
          putTheCallHere();
       }
    }
于 2020-07-26T07:50:23.633 回答
0

我有同样的问题，最新的 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
于 2019-03-23T04:16:19.273 回答