ND4J:生成范围内的随机数

ND4J: Generate random number within range

如果我们想创建一个 INDArray 的给定 填充 [=13= 之间的随机数] 我们可以使用:

Nd4j.rand(row, cols);

但是如果我们想要 INDArray 的给定 填充 -5000 and 5000 之间的随机数怎么办? 此外,如果每列 / vector 都有从特定范围内选择的随机数怎么办? 例如:

row = 2; columns = 3
pool = { {-500, 500}, {100, 100}, {1, 10} }
INDArray = [ [randRange(-500,500), randRange(100,100), randRange(1,10)],
             [randRange(-500,500), randRange(100,100), randRange(1,10)] ]

如何利用 Nd4j 的效率来实现这一点?

使用其他签名。

即这个:

Nd4j.rand(long[] shape, double min, double max, org.nd4j.linalg.api.rng.Random rng);

或者这个:

Nd4j.rand(INDArray target, double min, double max, org.nd4j.linalg.api.rng.Random rng);