Módulo

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

  • El tipo de entrada de la capa.

    Declaración

    associatedtype Input

  • El tipo de salida de la capa.

    Declaración

    associatedtype Output : Differentiable

  • Devuelve la salida obtenida al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    La salida.

  • hacia adelante(_:)

    Implementación predeterminada

    Devuelve la salida obtenida al aplicar la capa a la entrada dada.

    Implementación predeterminada

    Devuelve la salida obtenida al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    La salida.

Disponible donde `Input`:` TensorProtocol`, `Output`:` DifferentiableTensorProtocol`

  • callAsFunction (_ :)

    Implementación predeterminada

    Implementación predeterminada

    Devuelve la salida anotada obtenida al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    La salida anotada.

  • anotado (_ :)

    Método de extensión

    Anota output .

    Nota: las rentabilidades output si se utiliza un motor que no admite anotaciones.

    Declaración

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

    Parámetros

    output

    La salida a la capa.

    Valor devuelto

    La salida anotada.

  • resumen (entrada :)

    Método de extensión

    Devuelve las anotaciones obtenidas al aplicar la capa a la entrada dada.

    Declaración

    public func summary(input: Input) -> String

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    Todas las anotaciones recopiladas del gráfico XLA.