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 |
argmax(int... dimensions) |
SDVariable |
argmax(String name,
int... dimensions) |
SDVariable |
argmin(int... dimensions) |
SDVariable |
argmin(String name,
int... dimensions) |
SDVariable[] |
args()
Return the arguments for a given function
|
SDVariable |
assign(Number value) |
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
|
SDVariable |
eq(double value) |
SDVariable |
eq(SDVariable other) |
SDVariable |
eq(String name,
double value) |
SDVariable |
eq(String name,
SDVariable other) |
boolean |
equals(Object o) |
INDArray |
eval()
Evaluate the result of this variable
|
SDVariable |
get(SDIndex... indices) |
INDArray |
getArr()
A getter for the allocated ndarray with this
SDVariable . |
INDArray |
getArr(boolean enforceExistence)
A getter for the allocated ndarray with this
SDVariable . |
SDVariable |
getGradient()
A getter for the variable gradient.
|
long[] |
getShape()
Returns the shape of this variable
|
SDVariable |
gradient()
Nicer looking alias
for the gradient variable.
|
SDVariable |
gt(double value) |
SDVariable |
gt(SDVariable other) |
SDVariable |
gt(String name,
double value) |
SDVariable |
gt(String name,
SDVariable other) |
SDVariable |
gte(double value) |
SDVariable |
gte(SDVariable other) |
SDVariable |
gte(String name,
double value) |
SDVariable |
gte(String name,
SDVariable other) |
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 |
lt(double value) |
SDVariable |
lt(SDVariable other) |
SDVariable |
lt(String name,
double value) |
SDVariable |
lt(String name,
SDVariable other) |
SDVariable |
lte(double value) |
SDVariable |
lte(SDVariable other) |
SDVariable |
lte(String name,
double value) |
SDVariable |
lte(String name,
SDVariable other) |
SDVariable |
max(boolean keepDims,
int... dimensions) |
SDVariable |
max(int... dimensions) |
SDVariable |
max(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
max(String name,
int... dimensions) |
SDVariable |
mean(boolean keepDims,
int... dimensions) |
SDVariable |
mean(int... dimensions) |
SDVariable |
mean(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
mean(String name,
int... dimensions) |
SDVariable |
min(boolean keepDims,
int... dimensions) |
SDVariable |
min(int... dimensions) |
SDVariable |
min(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
min(String name,
int... dimensions) |
SDVariable |
mmul(SDVariable other) |
SDVariable |
mul(double sameDiffVariable) |
SDVariable |
mul(SDVariable sameDiffVariable) |
SDVariable |
mul(String varName,
double value) |
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) |
SDVariable |
neg()
Negate op
|
SDVariable |
neg(String name)
Negate op
|
SDVariable |
neq(double value) |
SDVariable |
neq(SDVariable other) |
SDVariable |
neq(String name,
double value) |
SDVariable |
neq(String name,
SDVariable other) |
SDVariable |
norm1(boolean keepDims,
int... dimensions) |
SDVariable |
norm1(int... dimensions) |
SDVariable |
norm1(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
norm1(String name,
int... dimensions) |
SDVariable |
norm2(boolean keepDims,
int... dimensions) |
SDVariable |
norm2(int... dimensions) |
SDVariable |
norm2(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
norm2(String name,
int... dimensions) |
SDVariable |
normmax(boolean keepDims,
int... dimensions) |
SDVariable |
normmax(int... dimensions) |
SDVariable |
normmax(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
normmax(String name,
int... dimensions) |
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 |
pow(double value) |
SDVariable |
pow(String varName,
double value) |
SDVariable |
prod(boolean keepDims,
int... dimensions) |
SDVariable |
prod(int... dimensions) |
SDVariable |
prod(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
prod(String name,
int... dimensions) |
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 |
setArray(INDArray array) |
SDVariable |
squaredDifference(SDVariable sameDiffVariable) |
SDVariable |
squaredDifference(String varName,
SDVariable sameDiffVariable) |
SDVariable |
std(boolean biasCorrected,
int... dimensions) |
SDVariable |
std(String name,
boolean biasCorrected,
boolean keepDims,
int... dimensions) |
SDVariable |
std(String name,
boolean biasCorrected,
int... dimensions) |
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) |
SDVariable |
sum(boolean keepDims,
int... dimensions) |
SDVariable |
sum(int... dimensions) |
SDVariable |
sum(String name,
boolean keepDims,
int... dimensions) |
SDVariable |
sum(String name,
int... dimensions) |
String |
tensorflowName()
The opName of this function tensorflow
|
String |
toString() |
SDVariable |
truncatedDiv(SDVariable sameDiffVariable) |
SDVariable |
truncatedDiv(String varName,
SDVariable sameDiffVariable) |
arg, argNames, asProperties, attributeAdaptersForFunction, calculateOutputShape, configFieldName, diff, f, getNumOutputs, getValue, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, opNum, outputVariable, outputVariablesNames, 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 this is possible. If this is not possible (due to shapes being unknown, etc)
null is returnedINDArray
associated with this variable.public INDArray getArr(boolean enforceExistence)
SDVariable
.
This getter will lazy initialize an array if one is not found based on the associated shape and
WeightInitScheme
- if this is possible.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 long[] getShape()
public SDVariable dup()
DifferentialFunction
dup
in class DifferentialFunction
public SDVariable assign(Number value)
public SDVariable neg()
public SDVariable neg(String name)
public SDVariable lt(double value)
public SDVariable lt(String name, double value)
public SDVariable lte(double value)
public SDVariable lte(String name, double value)
public SDVariable gt(double value)
public SDVariable gt(String name, double value)
public SDVariable gte(double value)
public SDVariable gte(String name, double value)
public SDVariable eq(double value)
public SDVariable eq(String name, double value)
public SDVariable neq(double value)
public SDVariable neq(String name, double value)
public SDVariable lt(SDVariable other)
public SDVariable lt(String name, SDVariable other)
public SDVariable lte(SDVariable other)
public SDVariable lte(String name, SDVariable other)
public SDVariable gt(SDVariable other)
public SDVariable gt(String name, SDVariable other)
public SDVariable gte(SDVariable other)
public SDVariable gte(String name, SDVariable other)
public SDVariable eq(SDVariable other)
public SDVariable eq(String name, SDVariable other)
public SDVariable neq(SDVariable other)
public SDVariable neq(String name, SDVariable other)
public SDVariable mmul(SDVariable other)
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 value)
value
- public SDVariable pow(double value)
public SDVariable pow(String varName, double value)
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 SDVariable sum(int... dimensions)
public SDVariable sum(boolean keepDims, int... dimensions)
public SDVariable sum(String name, int... dimensions)
public SDVariable sum(String name, boolean keepDims, int... dimensions)
public SDVariable mean(boolean keepDims, int... dimensions)
public SDVariable mean(String name, int... dimensions)
public SDVariable mean(int... dimensions)
public SDVariable mean(String name, boolean keepDims, int... dimensions)
public SDVariable std(boolean biasCorrected, int... dimensions)
public SDVariable std(String name, boolean biasCorrected, int... dimensions)
public SDVariable std(String name, boolean biasCorrected, boolean keepDims, int... dimensions)
public SDVariable prod(int... dimensions)
public SDVariable prod(boolean keepDims, int... dimensions)
public SDVariable prod(String name, int... dimensions)
public SDVariable prod(String name, boolean keepDims, int... dimensions)
public SDVariable min(int... dimensions)
public SDVariable min(boolean keepDims, int... dimensions)
public SDVariable min(String name, int... dimensions)
public SDVariable min(String name, boolean keepDims, int... dimensions)
public SDVariable max(int... dimensions)
public SDVariable max(boolean keepDims, int... dimensions)
public SDVariable max(String name, int... dimensions)
public SDVariable max(String name, boolean keepDims, int... dimensions)
public SDVariable norm1(int... dimensions)
public SDVariable norm1(boolean keepDims, int... dimensions)
public SDVariable norm1(String name, int... dimensions)
public SDVariable norm1(String name, boolean keepDims, int... dimensions)
public SDVariable norm2(int... dimensions)
public SDVariable norm2(boolean keepDims, int... dimensions)
public SDVariable norm2(String name, int... dimensions)
public SDVariable norm2(String name, boolean keepDims, int... dimensions)
public SDVariable normmax(int... dimensions)
public SDVariable normmax(boolean keepDims, int... dimensions)
public SDVariable normmax(String name, int... dimensions)
public SDVariable normmax(String name, boolean keepDims, int... dimensions)
public SDVariable argmax(int... dimensions)
public SDVariable argmax(String name, int... dimensions)
public SDVariable argmin(int... dimensions)
public SDVariable argmin(String name, int... dimensions)
public SDVariable setArray(INDArray array)
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
public SDVariable get(SDIndex... indices)
Copyright © 2018. All rights reserved.