@frozen
public struct Dilation2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Una capa de dilatación morfológica 2-D
Esta capa devuelve la dilatación morfágica del tensor de entrada con los filtros proporcionados.
El filtro de dilatación 4-D.
Declaración
public var filter: Tensor<Scalar>
Los pasos de la ventana deslizante para dimensiones espaciales.
Declaración
@noDerivative public let strides: (Int, Int)
El algoritmo de relleno para dilatación.
Declaración
@noDerivative public let padding: Padding
El factor de dilatación de las dimensiones espaciales.
Declaración
@noDerivative public let rates: (Int, Int)
Crea un
Dilation2D
capa con el filtro, zancadas, dilataciones y el relleno especificados.Declaración
Parámetros
filter
El filtro de dilatación 4-D de forma [altura del filtro, ancho del filtro, recuento de canales de entrada, recuento de canales de salida].
strides
Los pasos de la ventana deslizante para las dimensiones espaciales, es decir (altura de paso, ancho de paso).
rates
Las tasas de dilatación para las dimensiones espaciales, es decir (altura de dilatación, ancho de dilatación).
padding
El algoritmo de relleno para dilatación.
Devuelve la salida obtenida al aplicar la capa a la entrada dada.
Las dimensiones espaciales de salida se calculan como:
altura de salida = (altura de entrada + 2 * altura de relleno - (altura de dilatación * (altura de filtro - 1) + 1)) / altura de zancada + 1
ancho de salida = (ancho de entrada + 2 * ancho de relleno - (ancho de dilatación * (ancho de filtro - 1) + 1)) / ancho de zancada + 1
y los tamaños de acolchado están determinados por el esquema de acolchado.
Nota
Relleno tamaño es igual a cero cuando se utiliza
.valid
.Parámetros
input
La entrada a la capa de forma [tamaño de lote, altura de entrada, ancho de entrada, recuento de canales de entrada].
Valor devuelto
La salida de forma [recuento de lotes, altura de salida, ancho de salida, recuento de canales de salida].