Как поместить вектор в INDArray из ND4J?

Question

INDArray a = Nd4j.zeros(2,2);
a.put(1,Nd4j.create(nw float[]{1.0,2.0}));

Я пробовал выше код, но он дает ошибку ниже

java.lang.IllegalArgumentException: Element must be a scalar

Я новичок в ND4J lib. Есть ли способ напрямую поставить вектор?

 /**
     * Inserts the element at the specified index
     *
     * @param indices the indices to insert into
     * @param element a scalar ndarray
     * @return a scalar ndarray of the element at this index
     */
    INDArray put(int[] indices, INDArray element);

Я могу видеть выше метод в исходном коде, но для этого мне нужно будет создать массив int, который будет потреблять память.

Answer 1

Метод, который вы ищете putRow, который имеет желаемую подпись:

INDArray putRow(int row, INDArray toPut);

Существует также putColumn это работает аналогично для столбцов.