TraspostoConv2D

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

Uno strato di convoluzione trasposto 2-D (ad esempio, convoluzione spaziale trasposta sulle immagini).

Questo livello crea un filtro di convoluzione che viene trasposto-convoluto 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
  • La proprietà paddingIndex ci consente di gestire il calcolo basato sul riempimento.

    Dichiarazione

    @noDerivative
    public let paddingIndex: Int
  • Il tipo di funzione di attivazione per elemento.

    Dichiarazione

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>
  • Crea un livello TransposedConv2D con il filtro, il bias, la funzione di attivazione, gli step 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

    Un tensore di forma 4-D [height, width, output channel count, input channel count] .

    bias

    Il tensore di polarizzazione della forma [output channel count] .

    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 al livello.

    Valore di ritorno

    Il risultato.

  • Crea un livello TransposedConv2D 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

    Un tensore di forma 4-D [width, height, input channel count, output channel count] .

    strides

    I passi avanti della finestra scorrevole per le dimensioni spaziali.

    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.