Module

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

• Saisir
  Le type d’entrée du calque.

  Déclaration

  associatedtype Input

• Sortir
  Le type de sortie de la couche.

  Déclaration

  associatedtype Output : Differentiable

• appelAsFonction(_:)
  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

  La sortie.

• 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

  La sortie.

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.