Módulo

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

  • O tipo de entrada da camada.

    Declaração

    associatedtype Input

  • O tipo de saída da camada.

    Declaração

    associatedtype Output : Differentiable

  • Retorna a saída obtida da aplicação da camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.

  • frente(_:)

    Implementação padrão

    Retorna a saída obtida da aplicação da camada à entrada fornecida.

    Implementação Padrão

    Retorna a saída obtida da aplicação da camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.

Disponível onde `Input`:` TensorProtocol`, `Output`:` DifferentiableTensorProtocol`

  • callAsFunction (_ :)

    Implementação padrão

    Implementação Padrão

    Retorna a saída anotada obtida da aplicação da camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída anotada.

  • anotado (_ :)

    Método de extensão

    Anota output .

    NOTA: As devoluções output se estiver usando um backend que não suporta anotações.

    Declaração

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

    Parâmetros

    output

    A saída para a camada.

    Valor de retorno

    A saída anotada.

  • resumo (entrada :)

    Método de extensão

    Retorna as anotações obtidas da aplicação da camada à entrada fornecida.

    Declaração

    public func summary(input: Input) -> String

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    Todas as anotações coletadas do gráfico XLA.