dòng chảy :: Phạm vi

Đối tượng Scope đại diện cho một tập hợp các hoạt động TensorFlow liên quan có cùng thuộc tính, chẳng hạn như tiền tố tên chung.

Bản tóm tắt

Đối tượng Scope là một vùng chứa cho các thuộc tính TensorFlow Op. Các hàm tạo Op lấy một đối tượng Scope làm đối số đầu tiên bắt buộc và op được xây dựng nhận các thuộc tính trong đối tượng.

Một ví dụ đơn giản:

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()) { ... }

Phân cấp phạm vi :

Lớp Scope cung cấp các hàm With <> khác nhau để tạo một phạm vi mới. Phạm vi mới thường có một thuộc tính được thay đổi trong khi các thuộc tính khác được kế thừa từ phạm vi chính. Phương thức NewSubScope (name) gắn name vào tiền tố của tên cho các hoạt động được tạo trong phạm vi và WithOpName () thay đổi hậu tố mà nếu không thì mặc định thành loại op.

Ví dụ về tên:

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"

Phạm vi tồn tại:

Một phạm vi mới được tạo bằng cách gọi Scope :: NewRootScope . Điều này tạo ra một số tài nguyên được chia sẻ bởi tất cả các phạm vi con kế thừa từ phạm vi này, trực tiếp hoặc chuyển tiếp. Ví dụ: một phạm vi mới tạo ra một đối tượng Đồ thị mới mà các phép toán được thêm vào khi phạm vi mới hoặc phạm vi con của nó được sử dụng bởi một phương thức khởi tạo Op. Phạm vi mới cũng có một đối tượng Trạng thái sẽ được sử dụng để chỉ ra lỗi bởi các hàm Op-constructor được gọi trên bất kỳ phạm vi con nào. Các hàm Op-constructor phải kiểm tra trạng thái của phạm vi bằng cách gọi phương thức ok () trước khi tiến hành xây dựng op.

An toàn chủ đề:

Đối tượng Scope KHÔNG an toàn theo luồng. Các luồng không thể đồng thời gọi các hàm op-constructor trên cùng một đối tượng Scope .

Chức năng công cộng

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()