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.