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