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 and Description
BatchOutputConfig(@NonNull SameDiff sd)

Method Summary

Modifier and Type	Method and Description
Map<String,INDArray>	exec() Deprecated. Use output()
INDArray	execSingle() Deprecated. Use outputSingle()
BatchOutputConfig	input(@NonNull SDVariable variable, @NonNull INDArray placeholder) See input(String, INDArray)
BatchOutputConfig	input(@NonNull String variable, @NonNull 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
Map<String,INDArray>	output() Do inference and return the results
BatchOutputConfig	output(SDVariable... outputs) Add required outputs
BatchOutputConfig	output(String... 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

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
                                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
                               @NonNull String variable,
                               @NonNull
                               @NonNull INDArray 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)

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

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

Deprecated. Use output()

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