컨텍스트

public struct Context

레이어와 같은 딥러닝 API에서 사용하는 스레드-로컬 상황별 정보를 저장하는 컨텍스트입니다.

Context.local 사용하여 현재 스레드 로컬 컨텍스트를 검색합니다.

예:

  • BatchNorm 과 같은 레이어가 입력에 적용될 때 평균과 분산을 계산하도록 현재 학습 단계를 훈련으로 설정합니다.
  Context.local.learningPhase = .training
  • Dropout 과 같은 레이어가 입력에 적용될 때 단위를 삭제하지 않도록 현재 학습 단계를 추론으로 설정합니다.
  Context.local.learningPhase = .inference

  • 학습 단계.

    선언

    public var learningPhase: LearningPhase

  • 무작위 씨앗.

    메모

    무작위 시드는 획득될 때마다 업데이트되므로 향후 상태 비저장 무작위 TensorFlow 작업 실행으로 인해 비결정적 결과가 발생합니다.

    선언

    public var randomSeed: TensorFlowSeed { mutating get set }

  • 기본 속성을 사용하여 컨텍스트를 만듭니다.

    선언

    public init()

  • 현재 스레드 로컬 컨텍스트입니다.

    메모

    이 속성에 액세스하는 것은 스레드로부터 안전합니다.

    선언

    public static var local: Context { get set }