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.CustomOp
void clearArrays()
Copyright © 2020. All rights reserved.