| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.ops |
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
SDBaseOps.ifCond(SameDiffNoArgSingleLambda cond,
SameDiffNoArgSingleLambda trueBody,
SameDiffNoArgSingleLambda falseBody)
|
SDVariable |
SDBaseOps.ifCond(String ifName,
SameDiffNoArgSingleLambda cond,
SameDiffNoArgSingleLambda trueBody,
SameDiffNoArgSingleLambda falseBody)
|
SDVariable |
SDBaseOps.ifCond(String outputName,
String ifName,
SameDiffNoArgSingleLambda cond,
SameDiffNoArgSingleLambda trueBody,
SameDiffNoArgSingleLambda falseBody)
Constructs a If statement using the tensorflow style control flow operations (Switch and Merge)
If the result of cond is true, returns the result of trueBody, otherwise returns the result of falseBody
Note that cond and body lambdas are only called once to construct the graph.
|
Copyright © 2019. All rights reserved.