DepthwiseConv2D

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

Dwuwymiarowa warstwa splotu wgłębnego.

Warstwa ta tworzy rozdzielne filtry splotu, które są splatane z wejściem warstwy w celu wytworzenia tensora wyników.

  • Jądro splotu 4-D.

    Deklaracja

    public var filter: Tensor<Scalar>
  • Wektor odchylenia.

    Deklaracja

    public var bias: Tensor<Scalar>
  • Elementowa funkcja aktywacji.

    Deklaracja

    @noDerivative
    public let activation: Activation
  • Kroki okna przesuwnego dla wymiarów przestrzennych.

    Deklaracja

    @noDerivative
    public let strides: (Int, Int)
  • Algorytm dopełniania splotu.

    Deklaracja

    @noDerivative
    public let padding: Padding
  • Elementowy typ funkcji aktywacji.

    Deklaracja

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>
  • Tworzy warstwę DepthwiseConv2D z określonym filtrem, odchyleniem, funkcją aktywacji, krokami i dopełnieniem.

    Deklaracja

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

    Parametry

    filter

    Jądro splotu 4-D.

    bias

    Wektor odchylenia.

    activation

    Elementowa funkcja aktywacji.

    strides

    Kroki okna przesuwnego dla wymiarów przestrzennych.

    padding

    Algorytm dopełniania splotu.

  • Zwraca wynik uzyskany po zastosowaniu warstwy do podanych danych wejściowych.

    Deklaracja

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

    Parametry

    input

    Dane wejściowe do warstwy kształtu, [liczba partii, wysokość wejściowa, szerokość wejściowa, liczba kanałów wejściowych]

    Wartość zwracana

    Wynik kształtu, [liczba partii, wysokość wyjściowa, szerokość wyjściowa, liczba kanałów wejściowych * mnożnik kanału]

  • Tworzy warstwę DepthwiseConv2D z określonym kształtem filtra, krokami, dopełnieniem i funkcją aktywacji opartej na elementach.

    Deklaracja

    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()
    )

    Parametry

    filterShape

    Kształt jądra splotu 4-D o postaci [szerokość filtra, wysokość filtra, liczba kanałów wejściowych, mnożnik kanału].

    strides

    Kroki przesuwanego okna dla wymiarów przestrzennych/przestrzenno-czasowych.

    padding

    Algorytm dopełniania splotu.

    activation

    Elementowa funkcja aktywacji.

    filterInitializer

    Inicjator do użycia dla parametrów filtra.

    biasInitializer

    Inicjator do użycia dla parametrów odchylenia.