ГлубинаConv2D

@frozen
public struct DepthwiseConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

Двумерный слой глубинной свертки.

Этот слой создает отдельные фильтры свертки, которые свертываются с входными данными слоя для создания тензора выходных данных.

  • Ядро 4-D свертки.

    Декларация

    public var filter: Tensor<Scalar>

  • Вектор смещения.

    Декларация

    public var bias: Tensor<Scalar>

  • Функция поэлементной активации.

    Декларация

    @noDerivative
public let activation: Activation

  • Шаги скользящего окна для пространственных измерений.

    Декларация

    @noDerivative
public let strides: (Int, Int)

  • Алгоритм заполнения для свертки.

    Декларация

    @noDerivative
public let padding: Padding

  • Тип функции поэлементной активации.

    Декларация

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>

  • Создает слой DepthwiseConv2D с указанным фильтром, смещением, функцией активации, шагами и заполнением.

    Декларация

    public init(
  filter: Tensor<Scalar>,
  bias: Tensor<Scalar>? = nil,
  activation: @escaping Activation = identity,
  strides: (Int, Int) = (1, 1),
  padding: Padding = .valid
)

    Параметры

    filter

    Ядро 4-D свертки.

    bias

    Вектор смещения.

    activation

    Функция поэлементной активации.

    strides

    Шаги скользящего окна для пространственных измерений.

    padding

    Алгоритм заполнения для свертки.

  • Возвращает выходные данные, полученные в результате применения слоя к заданным входным данным.

    Декларация

    @differentiable
public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    Параметры

    input

    Ввод в слой формы, [количество пакетов, высота ввода, ширина ввода, количество входных каналов]

    Возвращаемое значение

    Форма вывода: [количество пакетов, высота вывода, ширина вывода, количество входных каналов * множитель канала]

  • Создает слой DepthwiseConv2D с указанной формой фильтра, шагами, заполнением и функцией активации поэлементно.

    Декларация

    public init(
  filterShape: (Int, Int, Int, Int),
  strides: (Int, Int) = (1, 1),
  padding: Padding = .valid,
  activation: @escaping Activation = identity,
  useBias: Bool = true,
  filterInitializer: ParameterInitializer<Scalar> = glorotUniform(),
  biasInitializer: ParameterInitializer<Scalar> = zeros()
)

    Параметры

    filterShape

    Форма четырехмерного ядра свертки с формой [ширина фильтра, высота фильтра, количество входных каналов, множитель канала].

    strides

    Шаги скользящего окна для пространственных/пространственно-временных измерений.

    padding

    Алгоритм заполнения для свертки.

    activation

    Функция поэлементной активации.

    filterInitializer

    Инициализатор, используемый для параметров фильтра.

    biasInitializer

    Инициализатор, используемый для параметров смещения.