Dilatation2D

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

Une couche de dilatation morphologique 2D

Cette couche renvoie la dilatation morphologique du tenseur d'entrée avec les filtres fournis

  • Le filtre de dilatation 4-D.

    Déclaration

    public var filter: Tensor<Scalar>
  • Les avancées de la fenêtre glissante pour les dimensions spatiales.

    Déclaration

    @noDerivative
    public let strides: (Int, Int)
  • L'algorithme de remplissage pour la dilatation.

    Déclaration

    @noDerivative
    public let padding: Padding
  • Le facteur de dilatation pour les dimensions spatiales.

    Déclaration

    @noDerivative
    public let rates: (Int, Int)
  • Crée un calque Dilation2D avec le filtre, les foulées, les dilatations et le remplissage spécifiés.

    Déclaration

    public init(
      filter: Tensor<Scalar>,
      strides: (Int, Int) = (1, 1),
      rates: (Int, Int) = (1, 1),
      padding: Padding = .valid
    )

    Paramètres

    filter

    Le filtre de dilatation 4D de forme [hauteur du filtre, largeur du filtre, nombre de canaux d'entrée, nombre de canaux de sortie].

    strides

    Les foulées de la fenêtre glissante pour les dimensions spatiales, c'est à dire (hauteur de foulée, largeur de foulée).

    rates

    Les taux de dilatation pour les dimensions spatiales, c'est-à-dire (hauteur de dilatation, largeur de dilatation).

    padding

    L'algorithme de remplissage pour la dilatation.

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

    Les dimensions spatiales en sortie sont calculées comme :

    hauteur de sortie = (hauteur d'entrée + 2 * hauteur de rembourrage - (hauteur de dilatation * (hauteur du filtre - 1) + 1)) / hauteur de foulée + 1

    largeur de sortie = (largeur d'entrée + 2 * largeur de rembourrage - (largeur de dilatation * (largeur du filtre - 1) + 1)) / largeur de foulée + 1

    et les tailles de remplissage sont déterminées par le schéma de remplissage.

    Note

    La taille du remplissage est égale à zéro lors de l'utilisation .valid .

    Déclaration

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

    Paramètres

    input

    L'entrée dans la couche de forme [taille du lot, hauteur d'entrée, largeur d'entrée, nombre de canaux d'entrée].

    Valeur de retour

    La sortie de la forme [nombre de lots, hauteur de sortie, largeur de sortie, nombre de canaux de sortie].