ProfonditàConv2D

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

Uno strato di convoluzione in profondità 2D.

Questo livello crea filtri di convoluzione separabili che vengono convoluti con l'input del livello per produrre un tensore di output.

  • Il kernel di convoluzione 4-D.

    Dichiarazione

    public var filter: Tensor<Scalar>

  • Il vettore di polarizzazione.

    Dichiarazione

    public var bias: Tensor<Scalar>

  • La funzione di attivazione per elemento.

    Dichiarazione

    @noDerivative
public let activation: Activation

  • I passi avanti della finestra scorrevole per le dimensioni spaziali.

    Dichiarazione

    @noDerivative
public let strides: (Int, Int)

  • L'algoritmo di riempimento per la convoluzione.

    Dichiarazione

    @noDerivative
public let padding: Padding

  • Il tipo di funzione di attivazione per elemento.

    Dichiarazione

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>

  • Crea un livello DepthwiseConv2D con il filtro, il bias, la funzione di attivazione, i passi e il riempimento specificati.

    Dichiarazione

    public init(
  filter: Tensor<Scalar>,
  bias: Tensor<Scalar>? = nil,
  activation: @escaping Activation = identity,
  strides: (Int, Int) = (1, 1),
  padding: Padding = .valid
)

    Parametri

    filter

    Il kernel di convoluzione 4-D.

    bias

    Il vettore di polarizzazione.

    activation

    La funzione di attivazione per elemento.

    strides

    I passi avanti della finestra scorrevole per le dimensioni spaziali.

    padding

    L'algoritmo di riempimento per la convoluzione.

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

    Dichiarazione

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

    Parametri

    input

    L'input per il livello di forma, [conteggio 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 input * moltiplicatore del canale]

  • Crea un livello DepthwiseConv2D con la forma del filtro, i passi, il riempimento e la funzione di attivazione per elemento specificati.

    Dichiarazione

    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()
)

    Parametri

    filterShape

    La forma del kernel di convoluzione 4-D con forma, [larghezza del filtro, altezza del filtro, conteggio dei canali di input, moltiplicatore del canale].

    strides

    I passi della finestra scorrevole per le dimensioni spazio/spazio-temporali.

    padding

    L'algoritmo di riempimento per la convoluzione.

    activation

    La funzione di attivazione per elemento.

    filterInitializer

    Inizializzatore da utilizzare per i parametri del filtro.

    biasInitializer

    Inizializzatore da utilizzare per i parametri di bias.