모듈

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

    레이어에 대한 입력입니다.

    반환 값

    출력.

`입력`: `TensorProtocol`, `출력`: `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 그래프에서 수집된 모든 주석입니다.