コンテキスト

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 }