Moduł 

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

  • Typ wejściowy warstwy.

    Deklaracja 

    associatedtype Input

  • Typ wyjściowy warstwy.

    Deklaracja 

    associatedtype Output : Differentiable

  • Zwraca wynik uzyskany po zastosowaniu warstwy do podanych danych wejściowych.

    Deklaracja 

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

    Parametry

    input

    Wejście do warstwy.

    Wartość zwracana

    Wyjście.

  • do przodu(_:) Domyślna implementacja

    Zwraca wynik uzyskany po zastosowaniu warstwy do podanych danych wejściowych.

    Domyślna implementacja

    Zwraca wynik uzyskany po zastosowaniu warstwy do podanych danych wejściowych.

    Deklaracja 

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

    Parametry

    input

    Wejście do warstwy.

    Wartość zwracana

    Wyjście.

Dostępne gdzie `Input`: `TensorProtocol`, `Output`: `DifferentiableTensorProtocol`
  • wywołajJakoFunkcję(_:) Domyślna implementacja

    Domyślna implementacja

    Zwraca wynik z adnotacjami uzyskany po zastosowaniu warstwy do danych wejściowych.

    Deklaracja 

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

    Parametry

    input

    Wejście do warstwy.

    Wartość zwracana

    Dane wyjściowe z adnotacjami.

  • z adnotacją(_:) Metoda przedłużania

    Adnotuje output .

    Uwaga: Zwraca output , jeśli używasz backendu, który nie obsługuje adnotacji.

    Deklaracja 

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

    Parametry

    output

    Dane wyjściowe do warstwy.

    Wartość zwracana

    Dane wyjściowe z adnotacjami.

  • podsumowanie(wejście:) Metoda przedłużania

    Zwraca adnotacje uzyskane w wyniku zastosowania warstwy do podanych danych wejściowych.

    Deklaracja 

    public func summary(input: Input) -> String

    Parametry

    input

    Wejście do warstwy.

    Wartość zwracana

    Wszystkie zebrane adnotacje z wykresu XLA.