Модуль

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

  • Входной тип слоя.

    Декларация

    associatedtype Input

  • Выходной тип слоя.

    Декларация

    associatedtype Output : Differentiable

  • Возвращает выходные данные, полученные в результате применения слоя к заданным входным данным.

    Декларация

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

    Параметры

    input

    Входные данные слоя.

    Возвращаемое значение

    Выход.

  • вперед(_:) Реализация по умолчанию

    Возвращает выходные данные, полученные в результате применения слоя к заданным входным данным.

    Реализация по умолчанию

    Возвращает выходные данные, полученные в результате применения слоя к заданным входным данным.

    Декларация

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

    Параметры

    input

    Входные данные слоя.

    Возвращаемое значение

    Выход.

Доступно, где `Input`: `TensorProtocol`, `Output`: `DifferentiableTensorProtocol`
  • callAsFunction(_:) Реализация по умолчанию

    Реализация по умолчанию

    Возвращает аннотированный вывод, полученный в результате применения слоя к заданным входным данным.

    Декларация

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

    Параметры

    input

    Входные данные слоя.

    Возвращаемое значение

    Аннотированный вывод.

  • аннотировано(_:) Метод расширения

    Аннотирует output .

    Примечание. Возвращает output , если используется серверная часть, не поддерживающая аннотации.

    Декларация

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

    Параметры

    output

    Вывод на слой.

    Возвращаемое значение

    Аннотированный вывод.

  • резюме (вход:) Метод расширения

    Возвращает аннотации, полученные в результате применения слоя к заданным входным данным.

    Декларация

    public func summary(input: Input) -> String

    Параметры

    input

    Входные данные слоя.

    Возвращаемое значение

    Все собранные аннотации из графа XLA.