SwiftCoreMLTools Documentation Beta

Structure Embedding

public struct Embedding: TrainableLayer
%3 Embedding Embedding TrainableLayer TrainableLayer Embedding->TrainableLayer

Conforms To

TrainableLayer

Initializers

init(name:​input:​output:​weight:​input​Dim:​output​Channels:​)

public init(name: String, input: [String], output: [String], weight: [Float], inputDim: UInt, outputChannels: UInt)

Properties

type

let type

name

let name: String

input

let input: [String]

output

let output: [String]

input​Dim

let inputDim: UInt

output​Channels

let outputChannels: UInt

weight

let weight: [Float]

bias

let bias: [Float]