Uses of Interface
org.nd4j.autodiff.samediff.SameDiffLambda
-
Packages that use SameDiffLambda Package Description org.nd4j.autodiff.samediff -
-
Uses of SameDiffLambda in org.nd4j.autodiff.samediff
Methods in org.nd4j.autodiff.samediff that return SameDiffLambda Modifier and Type Method Description static SameDiffLambdaControlFlow. loopBody(SameDiff parent, SameDiff functionBody, String functionName, String[] subGraphInputNames, String[] subGraphOutputNames)Create aSameDiffLambdato be used in combination withControlFlow.condBody()andSameDiff.invoke(Invoke.InvokeParams)this lambda will use samediff invoke as the function bdoy and setup the appropriate parameters to create a looping construct as described inControlFlow.loopBody(SameDiff, SameDiff, String, String[], String[])Methods in org.nd4j.autodiff.samediff with parameters of type SameDiffLambda Modifier and Type Method Description static SDVariable[]ControlFlow. whileLoop(SameDiff sameDiff, String[] outputNames, String loopName, @NonNull SDVariable[] loopVars, @NonNull SameDiffSingleLambda cond, @NonNull SameDiffLambda body)Constructs a While loop using the tensorflow style control flow operations (Switch, Merge, Enter, Exit, and NextIteration)SDVariable[]SameDiff. whileLoop(@NonNull SDVariable[] loopVars, @NonNull SameDiffSingleLambda cond, @NonNull SameDiffLambda body)SDVariable[]SameDiff. whileLoop(String[] outputNames, String loopName, @NonNull SDVariable[] loopVars, @NonNull SameDiffSingleLambda cond, @NonNull SameDiffLambda body)Constructs a While loop using the tensorflow style control flow operations (Switch, Merge, Enter, Exit, and NextIteration) Repeatedly executes body on the loop variables and updates them with the results, until cond evaluates to false Note that cond and body lambdas are only called once to construct the graph.SDVariable[]SameDiff. whileLoop(String loopName, @NonNull SDVariable[] loopVars, @NonNull SameDiffSingleLambda cond, @NonNull SameDiffLambda body)
-