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 © 2019. All rights reserved.