flux tenseur :: Portée

Un objet Scope représente un ensemble d'opérations TensorFlow associées qui ont les mêmes propriétés, comme un préfixe de nom commun.

Sommaire

Un objet Scope est un conteneur pour les propriétés TensorFlow Op. Les constructeurs d'op obtiennent un objet Scope comme premier argument obligatoire et l'op construit acquiert les propriétés de l'objet.

Un exemple simple :

using namespace ops;
Scope root = Scope::NewRootScope();
auto c1 = Const(root, { {1, 1} });
auto m = MatMul(root, c1, { {41}, {1} });
GraphDef gdef;
Status s = root.ToGraphDef(&gdef);
if (!s.ok()) { ... }

Hiérarchie de la portée :

La classe Scope fournit diverses fonctions With<> qui créent une nouvelle portée. La nouvelle étendue a généralement une propriété modifiée tandis que d'autres propriétés sont héritées de l'étendue parente. La méthode NewSubScope(name) ajoute name au préfixe des noms pour les opérations créées dans la portée, et WithOpName() modifie le suffixe qui, autrement, prend par défaut le type de l'opération.

Exemples de noms :

Scope root = Scope::NewRootScope();
Scope linear = root.NewSubScope("linear");
// W will be named "linear/W"
auto W = Variable(linear.WithOpName("W"),
                  {2, 2}, DT_FLOAT);
// b will be named "linear/b_3"
int idx = 3;
auto b = Variable(linear.WithOpName("b_", idx),
                  {2}, DT_FLOAT);
auto x = Const(linear, {...});  // name: "linear/Const"
auto m = MatMul(linear, x, W);  // name: "linear/MatMul"
auto r = BiasAdd(linear, m, b); // name: "linear/BiasAdd"

Durée de vie de la portée :

Une nouvelle portée est créée en appelant Scope::NewRootScope . Cela crée des ressources qui sont partagées par toutes les étendues enfants qui héritent de cette étendue, directement ou transitivement. Par exemple, une nouvelle portée crée un nouvel objet Graph auquel des opérations sont ajoutées lorsque la nouvelle portée ou ses enfants sont utilisés par un constructeur Op. La nouvelle portée a également un objet Status qui sera utilisé pour indiquer les erreurs par les fonctions du constructeur Op appelées sur n'importe quelle portée enfant. Les fonctions Op-constructor doivent vérifier l'état de la portée en appelant la méthode ok() avant de procéder à la construction de l'op.

Sécurité des fils :

Un objet Scope n'est PAS thread-safe. Les threads ne peuvent pas appeler simultanément des fonctions op-constructor sur le même objet Scope .

Fonctions publiques

Scope ClearColocation() const 

Scope ColocateWith(
  const Operation & op
) const 

Scope ColocateWith(
  const Output & out
) const 

Scope ExitOnError() const 

CompositeOpScopes GetCompositeOpScopes(
  const string & composite_op_name
) const 

string GetUniqueNameForOp(
  const string & default_name
) const 

Scope NewSubScope(
  const string & child_scope_name
) const 

 Scope(
  const Scope & other
)

Status ToGraphDef(
  GraphDef *gdef
) const 

void UpdateStatus(
  const Status & s
) const 

Scope WithAssignedDevice(
  const string & assigned_device
) const 

Scope WithControlDependencies(
  const gtl::ArraySlice< Operation > & control_deps
) const 

Scope WithControlDependencies(
  const Output & control_dep
) const 

Scope WithDevice(
  const string & device
) const 

Scope WithKernelLabel(
  const string & kernel_label
) const 

Scope WithNoControlDependencies() const 

Scope WithOpName(
  Ty... fragments
) const 

Scope WithXlaCluster(
  const string & xla_cluster
) const 

const std::vector< Operation > & control_deps() const 

Graph * graph() const 

std::shared_ptr< Graph > graph_as_shared_ptr() const 

bool ok() const 

Scope & operator=(
  const Scope & other
)

Status status() const 

 ~Scope()

Scope NewRootScope()