Дилатация2D

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

Двумерный морфологический слой расширения

Этот слой возвращает морфологическое расширение входного тензора с помощью предоставленных фильтров.

  • 4-D дилатационный фильтр.

    Декларация

    public var filter: Tensor<Scalar>
  • Шаги скользящего окна для пространственных измерений.

    Декларация

    @noDerivative
    public let strides: (Int, Int)
  • Алгоритм заполнения для расширения.

    Декларация

    @noDerivative
    public let padding: Padding
  • Коэффициент расширения пространственных измерений.

    Декларация

    @noDerivative
    public let rates: (Int, Int)
  • Создает слой Dilation2D с указанным фильтром, шагами, расширениями и отступами.

    Декларация

    public init(
      filter: Tensor<Scalar>,
      strides: (Int, Int) = (1, 1),
      rates: (Int, Int) = (1, 1),
      padding: Padding = .valid
    )

    Параметры

    filter

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

    strides

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

    rates

    Скорость расширения для пространственных измерений, т.е. (высота расширения, ширина расширения).

    padding

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

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

    Выходные пространственные размеры вычисляются как:

    высота вывода = (высота ввода + 2 * высота заполнения — (высота расширения * (высота фильтра — 1) + 1)) / высота шага + 1

    выходная ширина = (входная ширина + 2 * ширина заполнения - (ширина расширения * (ширина фильтра - 1) + 1)) / ширина шага + 1

    и размеры заполнения определяются схемой заполнения.

    Примечание

    Размер заполнения равен нулю при использовании .valid .

    Декларация

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

    Параметры

    input

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

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

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