| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff | |
| org.nd4j.autodiff.samediff.serde |
| Modifier and Type | Field and Description |
|---|---|
protected VariableType |
SDVariable.variableType |
| Modifier and Type | Method and Description |
|---|---|
static VariableType |
VariableType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static VariableType[] |
VariableType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
SameDiff.var(@NonNull String name,
@NonNull VariableType variableType,
WeightInitScheme weightInitScheme,
DataType dataType,
long... shape)
Variable initialization with a specified
WeightInitScheme
This method creates VARIABLE type SDVariable - i.e., must be floating point, and is a trainable parameter. |
| Constructor and Description |
|---|
SDVariable(@NonNull String varName,
@NonNull VariableType varType,
@NonNull SameDiff sameDiff,
long[] shape,
DataType dataType) |
| Modifier and Type | Method and Description |
|---|---|
static VariableType |
FlatBuffersMapper.fromVarType(byte varType) |
| Modifier and Type | Method and Description |
|---|---|
static byte |
FlatBuffersMapper.toVarType(VariableType variableType) |
Copyright © 2021. All rights reserved.