기준 치수

컬렉션을 사용해 정리하기 내 환경설정을 기준으로 콘텐츠를 저장하고 분류하세요.
  • 레이어의 입력 유형입니다.

    선언

    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 그래프에서 수집된 모든 주석.