Structure
Convolution
public struct Convolution: TrainableLayer
Relationships
Conforms To
TrainableLayer
Initializers
init(name:input:output:weight:bias:outputChannels:kernelChannels:nGroups:kernelSize:stride:dilationFactor:paddingType:outputShape: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]
outputChannels
let outputChannels: UInt
kernelChannels
let kernelChannels: UInt
nGroups
let nGroups: UInt
kernelSize
let kernelSize: [UInt]
stride
let stride: [UInt]
dilationFactor
let dilationFactor: [UInt]
paddingType
let paddingType: ConvolutionPaddingType
outputShape
let outputShape: [UInt]
deconvolution
let deconvolution: Bool
updatable
let updatable: Bool