0

如果我们想创建一个INDArray给定的之间填充随机数,0 and 1我们可以使用:

Nd4j.rand(row, cols);

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

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的效率来实现这一点?

4

1 回答 1

1

使用其他签名。

即这个:

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);

于 2020-04-30T09:42:11.763 回答