Package org.nd4j.autodiff.util
Class TrainingUtils
- java.lang.Object
-
- org.nd4j.autodiff.util.TrainingUtils
-
public class TrainingUtils extends Object
Utilities for SameDiff training and inference
-
-
Constructor Summary
Constructors Constructor Description TrainingUtils()
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static List<INDArray>
getSingleOutput(List<Map<String,INDArray>> outputs, String output)
Get a list of batch outputs for a single variable from a list of batch outputs for all variablesstatic Map<String,INDArray>
stackOutputs(List<Map<String,INDArray>> outputs)
Stack batch outputs, like an output fromSameDiff.output(MultiDataSetIterator, String...)
-