org.nd4j.autodiff.samediff.config

Class BatchOutputConfig

java.lang.Object

org.nd4j.autodiff.samediff.config.BatchOutputConfig

public class BatchOutputConfig
extends Object

Configuration for a single batch SameDiff inference operation. Used in SameDiff.batchOutput().

Constructor Summary

Constructors

Constructor and Description
BatchOutputConfig(SameDiff sd)

Method Summary

Modifier and Type	Method and Description
Map<String,INDArray>	exec() Do inference and return the results
INDArray	execSingle() Do inference and return the results for the single output Only works if exactly one output is specified
BatchOutputConfig	input(SDVariable variable, INDArray placeholder) See input(String, INDArray)
BatchOutputConfig	input(String variable, INDArray placeholder) Add a placeholder value for a specified variable
BatchOutputConfig	inputs(Map<String,INDArray> placeholders) Calls input(String, INDArray) on each entry in the map.
BatchOutputConfig	listeners(Listener... listeners) Add listeners for this operation
BatchOutputConfig	output(SDVariable... outputs) Add required outputs
BatchOutputConfig	output(String... outputs) Add required outputs
BatchOutputConfig	outputAll() Add all variables as required outputs

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

BatchOutputConfig

public BatchOutputConfig(@NonNull
                         SameDiff sd)

Method Detail

output

public BatchOutputConfig output(@NonNull
                                String... outputs)

Add required outputs

output

public BatchOutputConfig output(@NonNull
                                SDVariable... outputs)

Add required outputs

outputAll

public BatchOutputConfig outputAll()

Add all variables as required outputs

input

public BatchOutputConfig input(@NonNull
                               String variable,
                               @NonNull
                               INDArray placeholder)

Add a placeholder value for a specified variable

input

public BatchOutputConfig input(@NonNull
                               SDVariable variable,
                               @NonNull
                               INDArray placeholder)

See input(String, INDArray)

inputs

public BatchOutputConfig inputs(Map<String,INDArray> placeholders)

Calls input(String, INDArray) on each entry in the map.

listeners

public BatchOutputConfig listeners(@NonNull
                                   Listener... listeners)

Add listeners for this operation

exec

public Map<String,INDArray> exec()

Do inference and return the results

execSingle

public INDArray execSingle()

Do inference and return the results for the single output Only works if exactly one output is specified