org.nd4j.autodiff.samediff.optimize

Class OptimizationHelper

java.lang.Object

org.nd4j.autodiff.samediff.optimize.OptimizationHelper

public class OptimizationHelper
extends Object

Constructor Summary

Constructors

Constructor and Description
OptimizationHelper(SameDiff originalGraph, Properties properties)

Method Summary

Modifier and Type	Method and Description
OptimizationHelper	arrayRecoveryFunction(String arrayName, Supplier<INDArray> fn)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

OptimizationHelper

public OptimizationHelper(SameDiff originalGraph,
                          Properties properties)

Method Detail

arrayRecoveryFunction

public OptimizationHelper arrayRecoveryFunction(String arrayName,
                                                Supplier<INDArray> fn)