Dilatazione2D

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

Uno strato di dilatazione morfologica 2-D

Questo layer restituisce la dilatazione morfologica del tensore di input con i filtri forniti

  • Il filtro di dilatazione 4-D.

    Dichiarazione

    public var filter: Tensor<Scalar>
  • I passi avanti della finestra scorrevole per le dimensioni spaziali.

    Dichiarazione

    @noDerivative
    public let strides: (Int, Int)
  • L'algoritmo di riempimento per la dilatazione.

    Dichiarazione

    @noDerivative
    public let padding: Padding
  • Il fattore di dilatazione per le dimensioni spaziali.

    Dichiarazione

    @noDerivative
    public let rates: (Int, Int)
  • Crea un livello Dilation2D con il filtro, i passi, le dilatazioni e il riempimento specificati.

    Dichiarazione

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

    Parametri

    filter

    Il filtro di dilatazione 4-D della forma [altezza del filtro, larghezza del filtro, conteggio dei canali di ingresso, conteggio dei canali di uscita].

    strides

    I passi della finestra scorrevole per le dimensioni spaziali, ovvero (altezza del passo, larghezza del passo).

    rates

    I tassi di dilatazione per le dimensioni spaziali, ovvero (altezza di dilatazione, larghezza di dilatazione).

    padding

    L'algoritmo di riempimento per la dilatazione.

  • Restituisce l'output ottenuto dall'applicazione del livello all'input specificato.

    Le dimensioni spaziali di output sono calcolate come:

    altezza output = (altezza input + 2 * altezza imbottitura - (altezza dilatazione * (altezza filtro - 1) + 1)) / altezza passo + 1

    larghezza output = (larghezza input + 2 * larghezza imbottitura - (larghezza dilatazione * (larghezza filtro - 1) + 1)) / larghezza passo + 1

    e le dimensioni del riempimento sono determinate dallo schema di riempimento.

    Nota

    La dimensione del riempimento è uguale a zero quando si utilizza .valid .

    Dichiarazione

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

    Parametri

    input

    L'input al livello della forma [dimensione batch, altezza input, larghezza input, conteggio canali input].

    Valore di ritorno

    L'output della forma [conteggio batch, altezza di output, larghezza di output, conteggio dei canali di output].