Moy.Pool2D 

@frozen
public struct AvgPool2D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint

Une couche de regroupement moyenne pour les données spatiales.

  • Déclaration 

    public typealias TangentVector = EmptyTangentVector

  • La taille de la fenêtre de réduction glissante pour la mutualisation.

    Déclaration 

    @noDerivative
public let poolSize: (Int, Int, Int, Int)

  • Les foulées de la fenêtre coulissante pour chaque dimension d'une entrée 4D. Les progrès dans les dimensions non spatiales doivent être 1 .

    Déclaration 

    @noDerivative
public let strides: (Int, Int, Int, Int)

  • L'algorithme de remplissage pour le pooling.

    Déclaration 

    @noDerivative
public let padding: Padding

  • Crée une couche de pooling moyenne.

    Déclaration 

    public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)

  • Renvoie le résultat obtenu en appliquant le calque à l’entrée donnée.

    Déclaration 

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

    Paramètres

    input

    L'entrée dans le calque.

    Valeur de retour

    La sortie.

  • Crée une couche de pooling moyenne.

    Déclaration 

    public init(poolSize: (Int, Int), strides: (Int, Int), padding: Padding = .valid)

    Paramètres

    poolSize

    Facteurs verticaux et horizontaux permettant de réduire l'échelle.

    strides

    Les foulées.

    padding

    Le rembourrage.