View on GitHub
Install in Dash
Model Protocol Reference
Model
public protocol Model: GraphThis protocol defines higher-level APIs for creating, training and prediction of a model.
-
Operator symbols of this model
Declaration
Swift
var operators: [OperatorSymbol] -
List of input tensor symbols
Declaration
Swift
var inputs: [TensorSymbol] -
List of output tensor symbol
Declaration
Swift
var outputs: [TensorSymbol]
-
Add an input entry for moedel
inputShape
inputShapeindicates the shape of a sample without batch index. For example if we have some 128x128 RGB picture as input, theinputShapeshould beTensorShapeobject withshapeArray:[128, 128, 3]ifchannelOrderis ‘TensorChannelOrder.last’Declaration
Swift
func addInput(inputShape: TensorShape, channelOrder: TensorChannelOrder) -> TensorSymbolParameters
inputShapeinput shape of sample. Not include batch index.
Return Value
tensor symbol representation
-
Add layer to the model.
Declaration
Swift
func addLayer(_ inputs: [TensorSymbol], op: ComputableOperator) -> [TensorSymbol]Parameters
inputslist of input tensor symbols to this layer
opoperator
Return Value
list of output tensor symbols from this layer
-
Undocumented
Declaration
Swift
func configure()