Module

public protocol Module: EuclideanDifferentiable, KeyPathIterable
where
  TangentVector: VectorProtocol & ElementaryFunctions & PointwiseMultiplicative & KeyPathIterable

  • Le type d’entrée du calque.

    Déclaration

    associatedtype Input

  • Le type de sortie de la couche.

    Déclaration

    associatedtype Output : Differentiable

  • Renvoie le résultat obtenu 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 dans le calque.

    Valeur de retour

    Le résultat.

  • avant(_:) Implémentation par défaut

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

    Implémentation par défaut

    Renvoie le résultat obtenu 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 dans le calque.

    Valeur de retour

    Le résultat.

Disponible où `Input` : `TensorProtocol`, `Output` : `DifferentiableTensorProtocol`
  • appelAsFonction(_:) 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 dans le calque.

    Valeur de retour

    La sortie annotée.

  • annoté(_:) Méthode d'extension

    Annote output .

    Remarque : renvoie output si vous utilisez un backend qui ne prend pas en charge 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 dans le calque.

    Valeur de retour

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