View on GitHub
Install in Dash
Graph Protocol Reference
Graph
public protocol GraphBasic graph protocol
-
List of
GraphSymbol. Key is theUIDof symbol object in value field.Declaration
Swift
var symbols: [String: GraphSymbol] -
The readable label of this graph
Declaration
Swift
var graphLabel: String -
If this graph is trainable.
Declaration
Swift
var trainable: Bool -
Description of this graph
Declaration
Swift
var description: String -
Optimizer of this graph doing backward training. Could be
nilif just do forward calcualtion.Declaration
Swift
var optimizer: Optimizer? -
Counter of backward training
Declaration
Swift
var epoch: Int
-
Add a
TensorSymbolto the graph.Declaration
Swift
func tensor(_ label: String?, shape: TensorShape) -> TensorSymbolParameters
labellabel description
shapeshape description
Return Value
return value description
-
Add a
ScalarSymbolto the graph.Declaration
Swift
func scalar(_ label: String?, dataType: TensorDataType) -> ScalarSymbolParameters
labellabel
Return Value
-
Add a
OperatorSymbolto the graph.Declaration
Swift
func operation(_ label: String?, inputs: [TensorSymbol], op: ComputableOperator) -> (outputTensorSymbols: [TensorSymbol], operatorSymbol: OperatorSymbol, paramSymbols: [GraphSymbol])Parameters
inputsinput array of
TensorSymbolopA
ComputableOperatorinstanceReturn Value
Output
TensorSymbolfromoperatorcalculation, and constructedOperatorSymbol
-
Forward computing from inputs to outputs.
Declaration
Swift
func forward(mode: OperatorComputationMode)Parameters
modecomputation mode
-
Backward computing the grads for updatable data symbols.
Declaration
Swift
func backward(mode: OperatorComputationMode)Parameters
modecomputation mode
-
Bind data to
TensorSymbol.Declaration
Swift
func bindData(_ data: [String: DataSymbolSupportedDataType])Parameters
dataA dictinary whose key is
labelof aTensorSymboland value is an array ofDataSymbolSupportedDataTypeobjects. -
addSymbols(_:)Default implementationAdd symbol to
symbols. Should check duplicateDefault Implementation
Add symbol to
symbols. Should check duplicateDeclaration
Swift
func addSymbols(_ symbol: GraphSymbol)Parameters
symbolnew symbol