public interface Op
| Modifier and Type | Interface and Description |
|---|---|
static class |
Op.Type |
| Modifier and Type | Method and Description |
|---|---|
void |
clearArrays()
Clear the input and output INDArrays, if any are set
|
Object[] |
extraArgs()
Extra arguments
|
Buffer |
extraArgsBuff()
Returns a buffer of either float
or double
of the extra args for this buffer
|
DataBuffer |
extraArgsDataBuff(DataType bufferType)
Returns the extra args as a data buffer
|
String |
opName()
The opName of this operation
|
int |
opNum()
An op number
|
void |
setExtraArgs(Object[] extraArgs) |
void |
setX(INDArray x)
set x (the input ndarray)
|
void |
setY(INDArray y)
set y(the pairwise ndarray)
|
void |
setZ(INDArray z)
set z (the solution ndarray)
|
CustomOp |
toCustomOp()
|
INDArray |
x()
The origin ndarray
|
INDArray |
y()
The pairwise op ndarray
|
INDArray |
z()
The resulting ndarray
|
DataBuffer extraArgsDataBuff(DataType bufferType)
Buffer extraArgsBuff()
int opNum()
String opName()
INDArray x()
INDArray y()
INDArray z()
Object[] extraArgs()
void setX(INDArray x)
x - void setZ(INDArray z)
z - void setY(INDArray y)
y - void setExtraArgs(Object[] extraArgs)
extraArgs - CustomOp toCustomOp()
CustomOp
A CustomOp is a more flexible op
meant for multiple inputs and outputs.
The default implementation in BaseOp
converts a simple op to a multi input/output operation
by mapping the x and y on to inputs , the op opName
and the z on to outputs.CustomOpvoid clearArrays()
Copyright © 2021. All rights reserved.