Moy.Pool3D

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

Une couche de regroupement moyenne pour les données spatiales ou spatio-temporelles.

  • 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, Int)
  • Les foulées de la fenêtre coulissante pour chaque dimension d'une entrée 5D. Les progrès dans les dimensions non spatiales doivent être 1 .

    Déclaration

    @noDerivative
    public let strides: (Int, 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, Int),
      strides: (Int, 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, Int), strides: (Int, 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.

  • Crée une couche de pooling moyenne avec la taille de fenêtre de pooling et la foulée spécifiées. Toutes les tailles et foulées de pooling sont les mêmes.

    Déclaration

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