public class SDVariable extends DifferentialFunction implements Serializable
Modifier and Type | Field and Description |
---|---|
protected WeightInitScheme |
weightInitScheme |
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Modifier and Type | Method and Description |
---|---|
SDVariable |
add(double sameDiffVariable) |
SDVariable |
add(SDVariable sameDiffVariable) |
SDVariable |
add(String varName,
double sameDiffVariable) |
SDVariable |
add(String varName,
SDVariable sameDiffVariable) |
SDVariable |
addi(double sameDiffVariable) |
SDVariable |
addi(SDVariable sameDiffVariable) |
SDVariable |
addi(String varName,
double sameDiffVariable) |
SDVariable |
addi(String varName,
SDVariable sameDiffVariable) |
SDVariable |
arg()
Return the first argument
|
SDVariable[] |
args()
Return the arguments for a given function
|
SDVariable |
div(double sameDiffVariable) |
SDVariable |
div(SDVariable sameDiffVariable) |
SDVariable |
div(String varName,
double sameDiffVariable) |
SDVariable |
div(String varName,
SDVariable sameDiffVariable) |
SDVariable |
divi(double sameDiffVariable) |
SDVariable |
divi(SDVariable sameDiffVariable) |
SDVariable |
divi(String varName,
double sameDiffVariable) |
SDVariable |
divi(String varName,
SDVariable sameDiffVariable) |
List<SDVariable> |
doDiff(List<SDVariable> f1)
The actual implementation for automatic differentiation.
|
SDVariable |
dup()
Duplicate this function
|
boolean |
equals(Object o) |
INDArray |
eval()
Evaluate the result of this variable
|
INDArray |
getArr()
A getter for the allocated ndarray
with this
SDVariable . |
SDVariable |
getGradient()
A getter for the variable gradient.
|
int[] |
getShape()
Returns the shape of this variable
|
SDVariable |
gradient()
Nicer looking alias
for the gradient variable.
|
int |
hashCode() |
void |
initFromOnnx(OnnxProto3.NodeProto node,
SameDiff initWith,
Map<String,OnnxProto3.AttributeProto> attributesForNode,
OnnxProto3.GraphProto graph)
Iniitialize the function from the given
OnnxProto3.NodeProto |
void |
initFromTensorFlow(NodeDef nodeDef,
SameDiff initWith,
Map<String,AttrValue> attributesForNode,
GraphDef graph)
Initialize the function from the given
NodeDef |
boolean |
isPlaceHolder()
Returns true if this variable is a place holder
|
SDVariable |
mul(double sameDiffVariable) |
SDVariable |
mul(SDVariable sameDiffVariable) |
SDVariable |
mul(String varName,
double sameDiffVariable) |
SDVariable |
mul(String varName,
SDVariable sameDiffVariable) |
SDVariable |
muli(double sameDiffVariable) |
SDVariable |
muli(SDVariable sameDiffVariable) |
SDVariable |
muli(String varName,
double sameDiffVariable) |
SDVariable |
muli(String varName,
SDVariable sameDiffVariable) |
String |
onnxName()
The opName of this function in onnx
|
String |
opName()
The name of the op
|
Op.Type |
opType()
The type of the op
|
SDVariable[] |
outputVariables()
Return the output variables for this differential function.
|
SDVariable[] |
outputVariables(String baseName)
Return the output functions for this differential function.
|
SDVariable |
rdiv(double sameDiffVariable) |
SDVariable |
rdiv(SDVariable sameDiffVariable) |
SDVariable |
rdiv(String varName,
double sameDiffVariable) |
SDVariable |
rdiv(String varName,
SDVariable sameDiffVariable) |
SDVariable |
rdivi(double sameDiffVariable) |
SDVariable |
rdivi(SDVariable sameDiffVariable) |
SDVariable |
rdivi(String varName,
double sameDiffVariable) |
SDVariable |
rdivi(String varName,
SDVariable sameDiffVariable) |
SDVariable |
rsub(double sameDiffVariable) |
SDVariable |
rsub(SDVariable sameDiffVariable) |
SDVariable |
rsub(String varName,
double sameDiffVariable) |
SDVariable |
rsub(String varName,
SDVariable sameDiffVariable) |
SDVariable |
rsubi(double sameDiffVariable) |
SDVariable |
rsubi(SDVariable sameDiffVariable) |
SDVariable |
rsubi(String varName,
double sameDiffVariable) |
SDVariable |
rsubi(String varName,
SDVariable sameDiffVariable) |
SDVariable |
squaredDifference(SDVariable sameDiffVariable) |
SDVariable |
squaredDifference(String varName,
SDVariable sameDiffVariable) |
INDArray |
storeAndAllocateNewArray()
Allocate and return a new array
based on the vertex id and weight initialization.
|
SDVariable |
sub(double sameDiffVariable) |
SDVariable |
sub(SDVariable sameDiffVariable) |
SDVariable |
sub(String varName,
double sameDiffVariable) |
SDVariable |
sub(String varName,
SDVariable sameDiffVariable) |
SDVariable |
subi(double sameDiffVariable) |
SDVariable |
subi(SDVariable sameDiffVariable) |
SDVariable |
subi(String varName,
double sameDiffVariable) |
SDVariable |
subi(String varName,
SDVariable sameDiffVariable) |
String |
tensorflowName()
The opName of this function tensorflow
|
String |
toString() |
SDVariable |
truncatedDiv(SDVariable sameDiffVariable) |
SDVariable |
truncatedDiv(String varName,
SDVariable sameDiffVariable) |
asProperties, attributeAdaptersForFunction, calculateOutputShape, configFieldName, diff, f, getValue, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, opNum, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNames
protected WeightInitScheme weightInitScheme
public boolean isPlaceHolder()
public String opName()
DifferentialFunction
opName
in class DifferentialFunction
public SDVariable[] outputVariables()
DifferentialFunction
outputVariables
in class DifferentialFunction
public SDVariable arg()
DifferentialFunction
arg
in class DifferentialFunction
public SDVariable[] args()
DifferentialFunction
args
in class DifferentialFunction
public SDVariable[] outputVariables(String baseName)
DifferentialFunction
outputVariables
in class DifferentialFunction
public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)
DifferentialFunction
NodeDef
initFromTensorFlow
in class DifferentialFunction
public void initFromOnnx(OnnxProto3.NodeProto node, SameDiff initWith, Map<String,OnnxProto3.AttributeProto> attributesForNode, OnnxProto3.GraphProto graph)
DifferentialFunction
OnnxProto3.NodeProto
initFromOnnx
in class DifferentialFunction
public INDArray storeAndAllocateNewArray()
public INDArray getArr()
SDVariable
.
This getter will lazy initialize an array if one is not found
based on the associated shape and WeightInitScheme
if neither are found, an ND4JIllegalStateException
is thrown.
If a DifferentialFunction
is defined, note that
its getArr() method is called instead.INDArray
associated with this variable.public SDVariable gradient()
DifferentialFunction
public SDVariable getGradient()
public List<SDVariable> doDiff(List<SDVariable> f1)
DifferentialFunction
doDiff
in class DifferentialFunction
public int[] getShape()
public SDVariable dup()
DifferentialFunction
dup
in class DifferentialFunction
public SDVariable rsub(double sameDiffVariable)
sameDiffVariable
- public SDVariable rdiv(double sameDiffVariable)
sameDiffVariable
- public SDVariable add(double sameDiffVariable)
sameDiffVariable
- public SDVariable sub(double sameDiffVariable)
sameDiffVariable
- public SDVariable squaredDifference(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable div(double sameDiffVariable)
sameDiffVariable
- public SDVariable mul(double sameDiffVariable)
sameDiffVariable
- public SDVariable rsubi(double sameDiffVariable)
sameDiffVariable
- public SDVariable rdivi(double sameDiffVariable)
sameDiffVariable
- public SDVariable addi(double sameDiffVariable)
sameDiffVariable
- public SDVariable subi(double sameDiffVariable)
sameDiffVariable
- public SDVariable divi(double sameDiffVariable)
sameDiffVariable
- public SDVariable muli(double sameDiffVariable)
sameDiffVariable
- public SDVariable rsub(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rdiv(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable truncatedDiv(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable add(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable sub(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable div(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable mul(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rsubi(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rdivi(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable addi(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable subi(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable divi(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable muli(SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rsub(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable rdiv(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable truncatedDiv(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable add(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable sub(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable div(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable mul(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable rsubi(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable rdivi(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable addi(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable subi(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable divi(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable muli(String varName, double sameDiffVariable)
sameDiffVariable
- public SDVariable rsub(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rdiv(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable add(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable sub(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable squaredDifference(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable div(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable mul(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rsubi(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable rdivi(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable addi(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public Op.Type opType()
DifferentialFunction
opType
in class DifferentialFunction
public SDVariable subi(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable divi(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public SDVariable muli(String varName, SDVariable sameDiffVariable)
sameDiffVariable
- public INDArray eval()
public boolean equals(Object o)
equals
in class DifferentialFunction
public int hashCode()
hashCode
in class DifferentialFunction
public String onnxName()
DifferentialFunction
onnxName
in class DifferentialFunction
public String tensorflowName()
DifferentialFunction
tensorflowName
in class DifferentialFunction
Copyright © 2018. All rights reserved.