Как сохранить списки INDArray в наборе данных

Question

Для обучения подкреплению я создаю два List<INDArray> с состояниями и соответствующими предиктором движения и значением награды. Как я могу преобразовать эти два List в один org.nd4j.linalg.dataset.api.DataSet