Modifier and Type | Class and Description |
---|---|
static class |
ScatterUpdate.UpdateOp |
Constructor and Description |
---|
ScatterUpdate() |
ScatterUpdate(@NonNull INDArray original,
@NonNull INDArray updates,
INDArray result,
@NonNull int[] indices,
int[] dimension,
@NonNull ScatterUpdate.UpdateOp op) |
ScatterUpdate(@NonNull INDArray original,
@NonNull INDArray updates,
@NonNull int[] indices,
int[] dimension,
@NonNull ScatterUpdate.UpdateOp op) |
Modifier and Type | Method and Description |
---|---|
void |
addBArgument(boolean... arg) |
void |
addDArgument(DataType... arg) |
void |
addIArgument(int... arg) |
void |
addIArgument(long... 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 |
boolean[] |
bArgs() |
List<LongShapeDescriptor> |
calculateOutputShape()
Calculate the output shape for this op
|
List<LongShapeDescriptor> |
calculateOutputShape(OpContext opContext)
Calculate the output shape for this op
|
void |
clearArrays()
Clear the input and output INDArrays, if any are set
|
DataType[] |
dArgs() |
Boolean |
getBArgument(int index) |
CustomOpDescriptor |
getDescriptor()
Get the custom op descriptor if one is available.
|
Long |
getIArgument(int index) |
INDArray |
getInputArgument(int index) |
INDArray |
getOutputArgument(int index) |
Double |
getTArgument(int index) |
long[] |
iArgs() |
List<INDArray> |
inputArguments() |
boolean |
isInplaceCall()
This method returns true if op is supposed to be executed inplace
|
int |
numBArguments() |
int |
numDArguments() |
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
|
List<INDArray> |
outputArguments() |
void |
removeIArgument(Integer arg) |
void |
removeInputArgument(INDArray arg) |
void |
removeOutputArgument(INDArray arg) |
void |
removeTArgument(Double arg) |
double[] |
tArgs() |
protected CustomOp op
public ScatterUpdate()
public ScatterUpdate(@NonNull @NonNull INDArray original, @NonNull @NonNull INDArray updates, @NonNull @NonNull int[] indices, int[] dimension, @NonNull @NonNull ScatterUpdate.UpdateOp op)
public ScatterUpdate(@NonNull @NonNull INDArray original, @NonNull @NonNull INDArray updates, INDArray result, @NonNull @NonNull int[] indices, int[] dimension, @NonNull @NonNull ScatterUpdate.UpdateOp op)
public String opName()
public long opHash()
public boolean isInplaceCall()
isInplaceCall
in interface CustomOp
public List<INDArray> outputArguments()
outputArguments
in interface CustomOp
public List<INDArray> inputArguments()
inputArguments
in interface CustomOp
public void addIArgument(int... arg)
addIArgument
in interface CustomOp
public void addIArgument(long... arg)
addIArgument
in interface CustomOp
public void addBArgument(boolean... arg)
addBArgument
in interface CustomOp
public void removeIArgument(Integer arg)
removeIArgument
in interface CustomOp
public Boolean getBArgument(int index)
getBArgument
in interface CustomOp
public Long getIArgument(int index)
getIArgument
in interface CustomOp
public int numIArguments()
numIArguments
in interface CustomOp
public void addTArgument(double... arg)
addTArgument
in interface CustomOp
public void removeTArgument(Double arg)
removeTArgument
in interface CustomOp
public Double getTArgument(int index)
getTArgument
in interface CustomOp
public int numTArguments()
numTArguments
in interface CustomOp
public int numBArguments()
numBArguments
in interface CustomOp
public void addInputArgument(INDArray... arg)
addInputArgument
in interface CustomOp
public void removeInputArgument(INDArray arg)
removeInputArgument
in interface CustomOp
public INDArray getInputArgument(int index)
getInputArgument
in interface CustomOp
public int numInputArguments()
numInputArguments
in interface CustomOp
public void addOutputArgument(INDArray... arg)
addOutputArgument
in interface CustomOp
public void removeOutputArgument(INDArray arg)
removeOutputArgument
in interface CustomOp
public INDArray getOutputArgument(int index)
getOutputArgument
in interface CustomOp
public int numOutputArguments()
numOutputArguments
in interface CustomOp
public List<LongShapeDescriptor> calculateOutputShape()
CustomOp
calculateOutputShape
in interface CustomOp
public List<LongShapeDescriptor> calculateOutputShape(OpContext opContext)
CustomOp
calculateOutputShape
in interface CustomOp
public CustomOpDescriptor getDescriptor()
CustomOp
getDescriptor
in interface CustomOp
public void assertValidForExecution()
CustomOp
ND4JIllegalStateException
assertValidForExecution
in interface CustomOp
public void addDArgument(DataType... arg)
addDArgument
in interface CustomOp
public int numDArguments()
numDArguments
in interface CustomOp
public void clearArrays()
CustomOp
clearArrays
in interface CustomOp
Copyright © 2020. All rights reserved.