Package | Description |
---|---|
org.nd4j.autodiff.samediff.transform |
Modifier and Type | Method and Description |
---|---|
SubGraph |
SubGraphPredicate.getSubGraph(SameDiff sd,
DifferentialFunction rootFn)
Get the SubGraph that matches the predicate
|
Modifier and Type | Method and Description |
---|---|
static List<SubGraph> |
GraphTransformUtil.getSubgraphsMatching(SameDiff sd,
SubGraphPredicate p)
Get a list of all the subgraphs that match the specified predicate
|
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
SubGraphProcessor.processSubgraph(SameDiff sd,
SubGraph subGraph)
Replace the subgraph, and return the new outputs that should replace the old outputs.
Note that the order of the outputs you return matters! If the original outputs are [A,B,C] and you return output variables [X,Y,Z], then anywhere "A" was used as input will now use "X"; similarly Y replaces B, and Z replaces C. |
Copyright © 2020. All rights reserved.