@frozen
public struct SeparableConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Двумерный разделимый слой свертки.
Этот слой выполняет глубинную свертку, которая действует отдельно на каналы, за которой следует точечная свертка, которая смешивает каналы.
Ядро 4-D глубинной свертки.
Декларация
public var depthwiseFilter: Tensor<Scalar>
Ядро 4-D поточечной свертки.
Декларация
public var pointwiseFilter: Tensor<Scalar>
Вектор смещения.
Декларация
public var bias: Tensor<Scalar>
Поэлементная функция активации.
Декларация
@noDerivative public let activation: Activation
Шаги скользящего окна для пространственных измерений.
Декларация
@noDerivative public let strides: (Int, Int)
Алгоритм заполнения для свертки.
Декларация
@noDerivative public let padding: Padding
Создает слой
SeparableConv2D
с указанным глубинным и точечным фильтром, смещением, функцией активации, шагами и заполнением.Декларация
public init( depthwiseFilter: Tensor<Scalar>, pointwiseFilter: Tensor<Scalar>, bias: Tensor<Scalar>? = nil, activation: @escaping Activation = identity, strides: (Int, Int) = (1, 1), padding: Padding = .valid )
Параметры
depthwiseFilter
Ядро 4-D глубинной свертки
[filter height, filter width, input channels count, channel multiplier]
.pointwiseFilter
Ядро 4-D поточечной свертки
[1, 1, channel multiplier * input channels count, output channels count]
.bias
Вектор смещения.
activation
Поэлементная функция активации.
strides
Шаги скользящего окна для пространственных измерений.
padding
Алгоритм заполнения для свертки.
Создает слой
SeparableConv2D
с заданной формой фильтра по глубине и по точкам, шагами, заполнением и функцией активации поэлементно.Декларация
public init( depthwiseFilterShape: (Int, Int, Int, Int), pointwiseFilterShape: (Int, Int, Int, Int), strides: (Int, Int) = (1, 1), padding: Padding = .valid, activation: @escaping Activation = identity, useBias: Bool = true, depthwiseFilterInitializer: ParameterInitializer<Scalar> = glorotUniform(), pointwiseFilterInitializer: ParameterInitializer<Scalar> = glorotUniform(), biasInitializer: ParameterInitializer<Scalar> = zeros() )
Параметры
depthwiseFilterShape
Форма четырехмерного ядра глубинной свертки.
pointwiseFilterShape
Форма четырехмерного ядра точечной свертки.
strides
Шаги скользящего окна для пространственных/пространственно-временных измерений.
padding
Алгоритм заполнения для свертки.
activation
Функция поэлементной активации.
filterInitializer
Инициализатор, используемый для параметров фильтра.
biasInitializer
Инициализатор, используемый для параметров смещения.