Package 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	Description
OptimizationHelper(SameDiff originalGraph, Properties properties)

Method Summary

Modifier and Type	Method	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)