org.nd4j.linalg.api.ops.custom

Class ScatterUpdate

java.lang.Object

org.nd4j.linalg.api.ops.custom.ScatterUpdate

All Implemented Interfaces:

CustomOp

public class ScatterUpdate
extends Object
implements CustomOp

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ScatterUpdate.UpdateOp

Field Summary

Fields

Modifier and Type	Field and Description
protected CustomOp	op

Constructor Summary

Constructors

Constructor and Description
ScatterUpdate(INDArray original, INDArray updates, INDArray result, int[] indices, int[] dimension, ScatterUpdate.UpdateOp op)
ScatterUpdate(INDArray original, INDArray updates, int[] indices, int[] dimension, ScatterUpdate.UpdateOp op)

Method Summary

Modifier and Type	Method and Description
void	addIArgument(int... arg)
void	addInputArgument(INDArray... arg)
void	addOutputArgument(INDArray... arg)
void	addTArgument(double... arg)
void	assertValidForExecution() Asserts a valid state for execution, otherwise throws an ND4JIllegalStateException
List<int[]>	calculateOutputShape() Calculate the output shape for this op
CustomOpDescriptor	getDescriptor() Get the custom op descriptor if one is available.
Integer	getIArgument(int index)
INDArray	getInputArgument(int index)
INDArray	getOutputArgument(int index)
Double	getTArgument(int index)
int[]	iArgs()
INDArray[]	inputArguments()
boolean	isInplaceCall() This method returns true if op is supposed to be executed inplace
int	numIArguments()
int	numInputArguments()
int	numOutputArguments()
int	numTArguments()
long	opHash() This method returns LongHash of the opName()
String	opName() This method returns op opName as string
INDArray[]	outputArguments()
void	populateInputsAndOutputsFromSameDiff() Attempt to populate inputs and outputs from samediff.
void	removeIArgument(Integer arg)
void	removeInputArgument(INDArray arg)
void	removeOutputArgument(INDArray arg)
void	removeTArgument(Double arg)
double[]	tArgs()

Methods inherited from class java.lang.Object

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

Field Detail

op

protected CustomOp op

Constructor Detail

ScatterUpdate

public ScatterUpdate(@NonNull
                     INDArray original,
                     @NonNull
                     INDArray updates,
                     @NonNull
                     int[] indices,
                     int[] dimension,
                     @NonNull
                     ScatterUpdate.UpdateOp op)

ScatterUpdate

public ScatterUpdate(@NonNull
                     INDArray original,
                     @NonNull
                     INDArray updates,
                     INDArray result,
                     @NonNull
                     int[] indices,
                     int[] dimension,
                     @NonNull
                     ScatterUpdate.UpdateOp op)

Method Detail

opName

public String opName()

This method returns op opName as string

Specified by:

opName in interface CustomOp

Returns:

opHash

public long opHash()

This method returns LongHash of the opName()

Specified by:

opHash in interface CustomOp

Returns:

isInplaceCall

public boolean isInplaceCall()

This method returns true if op is supposed to be executed inplace

Specified by:

isInplaceCall in interface CustomOp

Returns:

outputArguments

public INDArray[] outputArguments()

Specified by:

outputArguments in interface CustomOp

inputArguments

public INDArray[] inputArguments()

Specified by:

inputArguments in interface CustomOp

iArgs

public int[] iArgs()

Specified by:

iArgs in interface CustomOp

tArgs

public double[] tArgs()

Specified by:

tArgs in interface CustomOp

addIArgument

public void addIArgument(int... arg)

Specified by:

addIArgument in interface CustomOp

removeIArgument

public void removeIArgument(Integer arg)

Specified by:

removeIArgument in interface CustomOp

getIArgument

public Integer getIArgument(int index)

Specified by:

getIArgument in interface CustomOp

numIArguments

public int numIArguments()

Specified by:

numIArguments in interface CustomOp

addTArgument

public void addTArgument(double... arg)

Specified by:

addTArgument in interface CustomOp

removeTArgument

public void removeTArgument(Double arg)

Specified by:

removeTArgument in interface CustomOp

getTArgument

public Double getTArgument(int index)

Specified by:

getTArgument in interface CustomOp

numTArguments

public int numTArguments()

Specified by:

numTArguments in interface CustomOp

addInputArgument

public void addInputArgument(INDArray... arg)

Specified by:

addInputArgument in interface CustomOp

removeInputArgument

public void removeInputArgument(INDArray arg)

Specified by:

removeInputArgument in interface CustomOp

getInputArgument

public INDArray getInputArgument(int index)

Specified by:

getInputArgument in interface CustomOp

numInputArguments

public int numInputArguments()

Specified by:

numInputArguments in interface CustomOp

addOutputArgument

public void addOutputArgument(INDArray... arg)

Specified by:

addOutputArgument in interface CustomOp

removeOutputArgument

public void removeOutputArgument(INDArray arg)

Specified by:

removeOutputArgument in interface CustomOp

getOutputArgument

public INDArray getOutputArgument(int index)

Specified by:

getOutputArgument in interface CustomOp

numOutputArguments

public int numOutputArguments()

Specified by:

numOutputArguments in interface CustomOp

calculateOutputShape

public List<int[]> calculateOutputShape()

Description copied from interface: CustomOp

Calculate the output shape for this op

Specified by:

calculateOutputShape in interface CustomOp

Returns:

getDescriptor

public CustomOpDescriptor getDescriptor()

Description copied from interface: CustomOp

Get the custom op descriptor if one is available.

Specified by:

getDescriptor in interface CustomOp

Returns:

assertValidForExecution

public void assertValidForExecution()

Description copied from interface: CustomOp

Asserts a valid state for execution, otherwise throws an ND4JIllegalStateException

Specified by:

assertValidForExecution in interface CustomOp

populateInputsAndOutputsFromSameDiff

public void populateInputsAndOutputsFromSameDiff()

Description copied from interface: CustomOp

Attempt to populate inputs and outputs from samediff. This method will not attempt to fill in integer or double aruguments.

Specified by:

populateInputsAndOutputsFromSameDiff in interface CustomOp