Контекст

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 }