Module

Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.
  • Le type d'entrée de la couche.

    Déclaration

    associatedtype Input
  • Le type de sortie de la couche.

    Déclaration

    associatedtype Output : Differentiable
  • Renvoie la sortie obtenue en appliquant le calque à l'entrée donnée.

    Déclaration

    @differentiable(wrt: self)
    func callAsFunction(_ input: Input) -> Output

    Paramètres

    input

    L'entrée de la couche.

    Valeur de retour

    Le résultat.

  • effronté(_:)

    Implémentation par défaut

    Renvoie la sortie obtenue en appliquant le calque à l'entrée donnée.

    Implémentation par défaut

    Renvoie la sortie obtenue en appliquant le calque à l'entrée donnée.

    Déclaration

    @differentiable(wrt: self)
    func forward(_ input: Input) -> Output

    Paramètres

    input

    L'entrée de la couche.

    Valeur de retour

    Le résultat.

Disponible où `Entrée` : `Protocole Tensor`, `Sortie` : `Protocole Tensor Différentiel`

  • callAsFunction(_:)

    Implémentation par défaut

    Implémentation par défaut

    Renvoie la sortie annotée obtenue en appliquant le calque à l'entrée donnée.

    Déclaration

    @differentiable(wrt: self)
    public func callAsFunction(_ input: Input) -> Output

    Paramètres

    input

    L'entrée de la couche.

    Valeur de retour

    La sortie annotée.

  • annoté(_:)

    Méthode d'extension

    Annote output .

    Note: retour output si vous utilisez un backend qui ne supporte pas les annotations.

    Déclaration

    @differentiable
    public func annotated(_ output: Output) -> Output

    Paramètres

    output

    La sortie vers la couche.

    Valeur de retour

    La sortie annotée.

  • résumé(entrée :)

    Méthode d'extension

    Renvoie les annotations obtenues en appliquant le calque à l'entrée donnée.

    Déclaration

    public func summary(input: Input) -> String

    Paramètres

    input

    L'entrée de la couche.

    Valeur de retour

    Toutes les annotations collectées à partir du graphique XLA.