SwiftCoreMLTools Documentation Beta

Structure Convolution

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

Conforms To

TrainableLayer

Initializers

init(name:​input:​output:​weight:​bias:​output​Channels:​kernel​Channels:​nGroups:​kernel​Size:​stride:​dilation​Factor:​padding​Type:​output​Shape:​deconvolution:​updatable:​)

public init(name: String, input: [String], output: [String], weight: [Float]? = nil, bias: [Float]? = nil, outputChannels: UInt, kernelChannels: UInt, nGroups: UInt, kernelSize: [UInt], stride: [UInt], dilationFactor: [UInt], paddingType: ConvolutionPaddingType, outputShape: [UInt], deconvolution: Bool, updatable: Bool = false)

Properties

type

let type

name

let name: String

input

let input: [String]

output

let output: [String]

weight

let weight: [Float]

bias

let bias: [Float]

output​Channels

let outputChannels: UInt

kernel​Channels

let kernelChannels: UInt

nGroups

let nGroups: UInt

kernel​Size

let kernelSize: [UInt]

stride

let stride: [UInt]

dilation​Factor

let dilationFactor: [UInt]

padding​Type

let paddingType: ConvolutionPaddingType

output​Shape

let outputShape: [UInt]

deconvolution

let deconvolution: Bool

updatable

let updatable: Bool