org.deeplearning4j.datasets.iterator.INDArrayDataSetIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-nn Show documentation

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.datasets.iterator;

import lombok.NonNull;
import org.nd4j.linalg.primitives.Pair;
import org.nd4j.linalg.api.ndarray.INDArray;

/**
 * @author [email protected]
 */
public class INDArrayDataSetIterator extends AbstractDataSetIterator {

    public INDArrayDataSetIterator(@NonNull Iterable> iterable, int batchSize) {
        super(iterable, batchSize);
    }
}