| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff |
| Modifier and Type | Method and Description |
|---|---|
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)
|
Copyright © 2021. All rights reserved.