@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
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
.Parametri
input
L'input al livello della forma [dimensione batch, altezza input, larghezza input, conteggio canali input].
Valore restituito
L'output della forma [conteggio batch, altezza di output, larghezza di output, conteggio dei canali di output].