org.nd4j.linalg.api.ops

Interface OpContext

All Superinterfaces:

AutoCloseable

All Known Implementing Classes:

BaseOpContext

public interface OpContext
extends AutoCloseable

This interface describes OpContext, abstraction used to setup op for execution.

Author:

[email protected]

Method Summary

Modifier and Type	Method and Description
org.bytedeco.javacpp.Pointer	contextPointer() This method returns pointer to context, to be used during native op execution
List<Boolean>	getBArguments()
List<Long>	getIArguments()
List<INDArray>	getInputArrays() This method returns List of input arrays defined within this context
List<INDArray>	getOutputArrays() This method returns List of output arrays defined within this context
org.nd4j.linalg.primitives.Pair<Long,Long>	getRngStates() This method returns RNG states, root first node second
List<Double>	getTArguments()
void	markInplace(boolean reallyInplace) This method allows to set op as inplace
void	setBArguments(boolean... arguments) This method sets boolean arguments required for operation
void	setIArguments(long... arguments) This method sets integer arguments required for operation
void	setInputArray(int index, INDArray array) This method adds INDArray as input argument for future op call
void	setInputArrays(INDArray... arrays) This method sets provided arrays as input arrays
void	setInputArrays(List<INDArray> arrays) This method sets provided arrays as input arrays
void	setOutputArray(int index, INDArray array) This method adds INDArray as output for future op call
void	setOutputArrays(INDArray... arrays) This method sets provided arrays as output arrays
void	setOutputArrays(List<INDArray> arrays) This method sets provided arrays as output arrays
void	setRngStates(long rootState, long nodeState) This method sets root-level seed for rng
void	setTArguments(double... arguments) This method sets floating point arguments required for operation

Methods inherited from interface java.lang.AutoCloseable

close

Method Detail

setIArguments

void setIArguments(long... arguments)

This method sets integer arguments required for operation

Parameters:

arguments -

getIArguments

List<Long> getIArguments()

setTArguments

void setTArguments(double... arguments)

This method sets floating point arguments required for operation

Parameters:

arguments -

getTArguments

List<Double> getTArguments()

setBArguments

void setBArguments(boolean... arguments)

This method sets boolean arguments required for operation

Parameters:

arguments -

getBArguments

List<Boolean> getBArguments()

setRngStates

void setRngStates(long rootState,
                  long nodeState)

This method sets root-level seed for rng

Parameters:

seed -

getRngStates

org.nd4j.linalg.primitives.Pair<Long,Long> getRngStates()

This method returns RNG states, root first node second

Returns:

setInputArray

void setInputArray(int index,
                   INDArray array)

This method adds INDArray as input argument for future op call

Parameters:

index -

array -

setInputArrays

void setInputArrays(List<INDArray> arrays)

This method sets provided arrays as input arrays

Parameters:

arrays -

setInputArrays

void setInputArrays(INDArray... arrays)

This method sets provided arrays as input arrays

Parameters:

arrays -

getInputArrays

List<INDArray> getInputArrays()

This method returns List of input arrays defined within this context

Returns:

setOutputArray

void setOutputArray(int index,
                    INDArray array)

This method adds INDArray as output for future op call

Parameters:

index -

array -

setOutputArrays

void setOutputArrays(List<INDArray> arrays)

This method sets provided arrays as output arrays

Parameters:

arrays -

setOutputArrays

void setOutputArrays(INDArray... arrays)

This method sets provided arrays as output arrays

Parameters:

arrays -

getOutputArrays

List<INDArray> getOutputArrays()

This method returns List of output arrays defined within this context

Returns:

contextPointer

org.bytedeco.javacpp.Pointer contextPointer()

This method returns pointer to context, to be used during native op execution

Returns:

markInplace

void markInplace(boolean reallyInplace)

This method allows to set op as inplace

Parameters:

reallyInplace -