org.nd4j.autodiff.samediff.config

Class OutputConfig

java.lang.Object

org.nd4j.autodiff.samediff.config.OutputConfig

public class OutputConfig
extends Object

Configuration for a SameDiff inference operation. Used in SameDiff.output().

Constructor Summary

Constructors

Constructor and Description
OutputConfig(SameDiff sd)

Method Summary

Modifier and Type	Method and Description
OutputConfig	data(DataSetIterator data) Set the data to use as input.
OutputConfig	data(MultiDataSetIterator 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(Listener... listeners) Add listeners for this operation
OutputConfig	output(SDVariable... outputs) Add required outputs
OutputConfig	output(String... 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
                    SameDiff sd)

Method Detail

output

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

Add required outputs

output

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

Add required outputs

data

public OutputConfig data(@NonNull
                         MultiDataSetIterator data)

Set the data to use as input.

data

public OutputConfig data(@NonNull
                         DataSetIterator data)

Set the data to use as input.

listeners

public OutputConfig listeners(@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.