@frozen
public struct SeparableConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPointUna capa de convolución separable en 2-D.
Esta capa realiza una convolución en profundidad que actúa por separado en los canales seguida de una convolución puntual que mezcla canales.
El núcleo de convolución profunda 4-D.
Declaración
public var depthwiseFilter: Tensor<Scalar>El núcleo de convolución puntual 4-D.
Declaración
public var pointwiseFilter: Tensor<Scalar>El vector de sesgo.
Declaración
public var bias: Tensor<Scalar>La función de activación por elementos.
Declaración
@noDerivative public let activation: ActivationLos pasos de la ventana corredera para las dimensiones espaciales.
Declaración
@noDerivative public let strides: (Int, Int)El algoritmo de relleno para la convolución.
Declaración
@noDerivative public let padding: PaddingCrea una capa
SeparableConv2Dcon el filtro en profundidad y en puntos, el sesgo, la función de activación, los avances y el relleno especificados.Declaración
public init( depthwiseFilter: Tensor<Scalar>, pointwiseFilter: Tensor<Scalar>, bias: Tensor<Scalar>? = nil, activation: @escaping Activation = identity, strides: (Int, Int) = (1, 1), padding: Padding = .valid )Parámetros
depthwiseFilterEl núcleo de convolución en profundidad 4-D
[filter height, filter width, input channels count, channel multiplier].pointwiseFilterEl núcleo de convolución puntual 4-D
[1, 1, channel multiplier * input channels count, output channels count].biasEl vector de sesgo.
activationLa función de activación por elementos.
stridesLos pasos de la ventana corredera para las dimensiones espaciales.
paddingEl algoritmo de relleno para la convolución.
Crea una capa
SeparableConv2Dcon la forma de filtro puntual y en profundidad, los pasos, el relleno y la función de activación de elementos especificados.Declaración
public init( depthwiseFilterShape: (Int, Int, Int, Int), pointwiseFilterShape: (Int, Int, Int, Int), strides: (Int, Int) = (1, 1), padding: Padding = .valid, activation: @escaping Activation = identity, useBias: Bool = true, depthwiseFilterInitializer: ParameterInitializer<Scalar> = glorotUniform(), pointwiseFilterInitializer: ParameterInitializer<Scalar> = glorotUniform(), biasInitializer: ParameterInitializer<Scalar> = zeros() )Parámetros
depthwiseFilterShapeLa forma del núcleo de convolución en profundidad 4-D.
pointwiseFilterShapeLa forma del núcleo de convolución puntual 4-D.
stridesLos pasos de la ventana corredera para las dimensiones espacio/espacio-temporales.
paddingEl algoritmo de relleno para la convolución.
activationLa función de activación por elementos.
filterInitializerInicializador que se utilizará para los parámetros del filtro.
biasInitializerInicializador que se utilizará para los parámetros de polarización.