| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff |
| Modifier and Type | Method and Description |
|---|---|
OutAndGrad |
SameDiff.calculateGradientsAndOutputs(Map<String,INDArray> placeholderVals,
Collection<String> outputVars,
Collection<String> gradientVars)
Calculate the activations and the gradients for the specified variables, in one execution call.
|
Copyright © 2021. All rights reserved.