| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff | |
| org.nd4j.autodiff.samediff.config |
| Modifier and Type | Method and Description |
|---|---|
BatchOutputConfig |
SameDiff.batchOutput()
Set up for a single batch inference operation using OutputConfig.
|
| Modifier and Type | Method and Description |
|---|---|
BatchOutputConfig |
BatchOutputConfig.input(@NonNull SDVariable variable,
@NonNull INDArray placeholder)
|
BatchOutputConfig |
BatchOutputConfig.input(@NonNull String variable,
@NonNull INDArray placeholder)
Add a placeholder value for a specified variable
|
BatchOutputConfig |
BatchOutputConfig.inputs(Map<String,INDArray> placeholders)
Calls
input(String, INDArray) on each entry in the map. |
BatchOutputConfig |
BatchOutputConfig.listeners(Listener... listeners)
Add listeners for this operation
|
BatchOutputConfig |
BatchOutputConfig.output(SDVariable... outputs)
Add required outputs
|
BatchOutputConfig |
BatchOutputConfig.output(String... outputs)
Add required outputs
|
BatchOutputConfig |
BatchOutputConfig.outputAll()
Add all variables as required outputs
|
Copyright © 2021. All rights reserved.