Package org.nd4j.autodiff.samediff
Class ControlFlow.LoopLambdaArgs
- java.lang.Object
-
- org.nd4j.autodiff.samediff.ControlFlow.LoopLambdaArgs
-
- Enclosing class:
- ControlFlow
public static class ControlFlow.LoopLambdaArgs extends Object
-
-
Constructor Summary
Constructors Constructor Description LoopLambdaArgs(SDVariable iterStart, SDVariable iterCount, SDVariable[] extraArgs, SDVariable condIn)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Invoke.InvokeParams
invokeParams(String functionName, String[] subGraphInputNames, String[] subGraphOutputNames)
ConstructInvoke.InvokeParams
for usage withSameDiff.invoke(Invoke.InvokeParams)
the variables here reflect what is used in the loop.
-
-
-
Constructor Detail
-
LoopLambdaArgs
public LoopLambdaArgs(SDVariable iterStart, SDVariable iterCount, SDVariable[] extraArgs, SDVariable condIn)
-
-
Method Detail
-
invokeParams
public Invoke.InvokeParams invokeParams(String functionName, String[] subGraphInputNames, String[] subGraphOutputNames)
ConstructInvoke.InvokeParams
for usage withSameDiff.invoke(Invoke.InvokeParams)
the variables here reflect what is used in the loop. A user can useControlFlow.LoopLambdaArgs
to create an appropriately configuredInvoke.InvokeParams
to be used with the body.- Parameters:
functionName
- the name of the function to invokesubGraphInputNames
- the subgraph input names to invoke the function withsubGraphOutputNames
- the subgraph output names to expect returned from the function- Returns:
- the appropriate invoke parameters for use with
ControlFlow.condBody()
andControlFlow.loopBody(SameDiff, SameDiff, String, String[], String[])
-
-