Package org.nd4j.autodiff.samediff.config

Class BatchOutputConfig

java.lang.Object

org.nd4j.autodiff.samediff.config.BatchOutputConfig

public class BatchOutputConfig
extends Object

Constructor Summary

Constructors

Constructor	Description
BatchOutputConfig(@NonNull SameDiff sd)

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Detail

BatchOutputConfig

public BatchOutputConfig(@NonNull
                         @NonNull SameDiff sd)

Method Detail

output

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

Add required outputs

output

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

Add required outputs

outputAll

public BatchOutputConfig outputAll()

Add all variables as required outputs

input

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

Add a placeholder value for a specified variable

valueInput

public BatchOutputConfig valueInput(@NonNull
                                    @NonNull String variable,
                                    @NonNull
                                    @NonNull SDValue placeholder)

Add a placeholder value for a specified variable

input

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

See input(String, INDArray)

valueInputs

public BatchOutputConfig valueInputs(Map<String,SDValue> placeholders)

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

inputs

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

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

listeners

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

Add listeners for this operation

exec

@Deprecated
public Map<String,INDArray> exec()

Deprecated.

Use output()

outputValue

public Map<String,SDValue> outputValue()

Do inference and return the results

output

public Map<String,INDArray> output()

Do inference and return the results

execSingle

@Deprecated
public INDArray execSingle()

Deprecated.

Use outputSingle()

outputSingle

public INDArray outputSingle()

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