Google I/O est terminé ! Suivez les sessions TensorFlow Afficher les sessions

Le contexte

public struct Context

Un contexte qui stocke les informations contextuelles locales au thread utilisées par les API d'apprentissage en profondeur telles que les couches.

Utilisez Context.local pour récupérer le contexte local des threads en cours.

Exemples:

  • Régler la phase d'apprentissage en cours de formation afin que les couches comme BatchNorm calculeront moyenne et la variance lorsqu'il est appliqué aux entrées.
  Context.local.learningPhase = .training
  • Réglez la phase d'apprentissage en cours à l' inférence de sorte que les couches comme Dropout ne retombera unités lorsqu'elles sont appliquées aux entrées.
  Context.local.learningPhase = .inference
  • La phase d'apprentissage.

    Déclaration

    public var learningPhase: LearningPhase
  • La graine aléatoire.

    Noter

    Chaque fois qu'elle est obtenue, la graine aléatoire est également mise à jour afin que les futures exécutions d'opérations TensorFlow aléatoires sans état se traduisent par des résultats non déterministes.

    Déclaration

    public var randomSeed: TensorFlowSeed { mutating get set }
  • Crée un contexte avec des propriétés par défaut.

    Déclaration

    public init()
  • Le contexte actuel du thread local.

    Noter

    L'accès à cette propriété est thread-safe.

    Déclaration

    public static var local: Context { get set }