Package org.nd4j.autodiff.samediff.config

Class OutputConfig

java.lang.Object

org.nd4j.autodiff.samediff.config.OutputConfig

public class OutputConfig
extends Object

Constructor Summary

Constructors

Constructor	Description
OutputConfig(@NonNull SameDiff sd)

Method Summary

Modifier and Type	Method	Description
OutputConfig	data(@NonNull DataSet data)	Set the data to use as input.
OutputConfig	data(@NonNull DataSetIterator data)	Set the data to use as input.
OutputConfig	data(@NonNull MultiDataSetIterator data)	Set the data to use as input.
OutputConfig	data(@NonNull MultiDataSet data)	Set the data to use as input.
Map<String,INDArray>	exec()	Do inference and return the results.
List<Map<String,INDArray>>	execBatches()	Do inference and return the results in batches.
INDArray	execSingle()	Do inference and return the results for the single output variable specified.
List<INDArray>	execSingleBatches()	Do inference and return the results (in batches) for the single output variable specified.
OutputConfig	listeners(@NonNull Listener... listeners)	Add listeners for this operation
OutputConfig	output(@NonNull String... outputs)	Add required outputs
OutputConfig	output(@NonNull SDVariable... outputs)	Add required outputs

Methods inherited from class java.lang.Object

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

Constructor Detail

OutputConfig

public OutputConfig(@NonNull
                    @NonNull SameDiff sd)

Method Detail

output

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

Add required outputs

output

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

Add required outputs

data

public OutputConfig data(@NonNull
                         @NonNull MultiDataSetIterator data)

Set the data to use as input.

data

public OutputConfig data(@NonNull
                         @NonNull DataSetIterator data)

Set the data to use as input.

data

public OutputConfig data(@NonNull
                         @NonNull DataSet data)

Set the data to use as input.

data

public OutputConfig data(@NonNull
                         @NonNull MultiDataSet data)

Set the data to use as input.

listeners

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

Add listeners for this operation

exec

public Map<String,INDArray> exec()

Do inference and return the results. Uses concatenation on the outputs of execBatches() which may cause issues with some inputs. RNNs with variable time series length and CNNs with variable image sizes will most likely have issues.

execBatches

public List<Map<String,INDArray>> execBatches()

Do inference and return the results in batches.

execSingle

public INDArray execSingle()

Do inference and return the results for the single output variable specified. Only works if exactly one output is specified. Uses concatenation on the outputs of execBatches() which may cause issues with some inputs. RNNs with variable time series length and CNNs with variable image sizes will most likely have issues.

execSingleBatches

public List<INDArray> execSingleBatches()

Do inference and return the results (in batches) for the single output variable specified. Only works if exactly one output is specified.