Как выбрать случайным образом элементы INDrray Nd4j с равномерным распределением?

Question

С numpy я бы сделал это:

arr = np.arange(10)
res = np.random.choice(arr, size=(3,3))

Если параметр p (вероятности) не указан, то распределение является равномерным. Однако в Nd4j, требуется параметр вероятностей .

Должен ли я сам строить параметр вероятностей или есть более подходящий способ?

INDArray arr = Nd4j.create(new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9});
INDArray res = Nd4j.zeroes(3, 3);
Nd4j.choice(arr, ..., res);