dòng chảy căng:: Phiên khách hàng
#include <client_session.h>
Đối tượng ClientSession
cho phép người gọi điều khiển việc đánh giá biểu đồ TensorFlow được xây dựng bằng API C++.
Bản tóm tắt
Ví dụ:
Scope root = Scope::NewRootScope(); auto a = Placeholder(root, DT_INT32); auto c = Add(root, a, {41}); ClientSession session(root); std::vectoroutputs; Status s = session.Run({ {a, {1} } }, {c}, &outputs); if (!s.ok()) { ... }
Hàm tạo và hàm hủy | |
---|---|
ClientSession (const Scope & scope, const string & target) Tạo một phiên mới để đánh giá biểu đồ có trong scope bằng cách kết nối với thời gian chạy TensorFlow được chỉ định bởi target . | |
ClientSession (const Scope & scope) Tương tự như trên, nhưng sử dụng chuỗi trống ("") làm thông số đích. | |
ClientSession (const Scope & scope, const SessionOptions & session_options) Tạo một phiên mới, định cấu hình nó với session_options . | |
~ClientSession () |
Các loại công khai | |
---|---|
CallableHandle | typedefint64 Một mã điều khiển cho biểu đồ con, được tạo bằng ClientSession::MakeCallable() . |
FeedType | typedefstd::unordered_map< Output , Input::Initializer , OutputHash > Loại dữ liệu đại diện cho nguồn cấp dữ liệu cho lệnh gọi Chạy. |
Chức năng công cộng | |
---|---|
MakeCallable (const CallableOptions & callable_options, CallableHandle *out_handle) | Status Tạo một handle để gọi sơ đồ con được xác định bởi callable_options . |
ReleaseCallable ( CallableHandle handle) | Status Giải phóng các tài nguyên được liên kết với handle đã cho trong phiên này. |
Run (const std::vector< Output > & fetch_outputs, std::vector< Tensor > *outputs) const | Status Đánh giá các tensor fetch_outputs . |
Run (const FeedType & inputs, const std::vector< Output > & fetch_outputs, std::vector< Tensor > *outputs) const | Status Tương tự như trên, nhưng sử dụng ánh xạ trong inputs làm nguồn cấp dữ liệu. |
Run (const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs) const | Status Tương tự như trên. Ngoài ra còn chạy các hoạt động trong run_outputs . |
Run (const RunOptions & run_options, const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs, RunMetadata *run_metadata) const | Status Sử dụng run_options để bật hồ sơ hiệu suất. |
Run (const RunOptions & run_options, const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs, RunMetadata *run_metadata, const thread::ThreadPoolOptions & threadpool_options) const | Status Tương tự như trên. |
RunCallable ( CallableHandle handle, const std::vector< Tensor > & feed_tensors, std::vector< Tensor > *fetch_tensors, RunMetadata *run_metadata) | Status Gọi đồ thị con được đặt tên bằng handle với các tùy chọn đã cho và các tensor đầu vào. |
RunCallable ( CallableHandle handle, const std::vector< Tensor > & feed_tensors, std::vector< Tensor > *fetch_tensors, RunMetadata *run_metadata, const thread::ThreadPoolOptions & options) | Status Gọi đồ thị con được đặt tên bằng handle với các tùy chọn đã cho và các tensor đầu vào. |
Các loại công khai
Xử lý có thể gọi được
int64 CallableHandle
Một mã điều khiển cho biểu đồ con, được tạo bằng ClientSession::MakeCallable()
.
Loại nguồn cấp dữ liệu
std::unordered_map< Output, Input::Initializer, OutputHash > FeedType
Loại dữ liệu đại diện cho nguồn cấp dữ liệu cho lệnh gọi Chạy.
Đây là bản đồ của các đối tượng Output
các hàm tạo op trả về giá trị để cung cấp cho chúng. Xem Input::Initializer
để biết chi tiết về những gì có thể được sử dụng làm giá trị nguồn cấp dữ liệu.
Chức năng công cộng
Phiên khách hàng
ClientSession( const Scope & scope, const string & target )
Tạo một phiên mới để đánh giá biểu đồ có trong scope
bằng cách kết nối với thời gian chạy TensorFlow được chỉ định bởi target
.
Phiên khách hàng
ClientSession( const Scope & scope )
Tương tự như trên, nhưng sử dụng chuỗi trống ("") làm thông số đích.
Phiên khách hàng
ClientSession( const Scope & scope, const SessionOptions & session_options )
Tạo một phiên mới, định cấu hình nó với session_options
.
MakeCallable
Status MakeCallable( const CallableOptions & callable_options, CallableHandle *out_handle )
Tạo một handle
để gọi sơ đồ con được xác định bởi callable_options
.
LƯU Ý: API này vẫn đang thử nghiệm và có thể thay đổi.
Phát hành có thể gọi được
Status ReleaseCallable( CallableHandle handle )
Giải phóng các tài nguyên được liên kết với handle
đã cho trong phiên này.
LƯU Ý: API này vẫn đang thử nghiệm và có thể thay đổi.
Chạy
Status Run( const std::vector< Output > & fetch_outputs, std::vector< Tensor > *outputs ) const
Đánh giá các tensor fetch_outputs
.
Các giá trị được trả về dưới dạng đối tượng Tensor
trong outputs
. Số lượng và thứ tự outputs
sẽ khớp với fetch_outputs
.
Chạy
Status Run( const FeedType & inputs, const std::vector< Output > & fetch_outputs, std::vector< Tensor > *outputs ) const
Tương tự như trên, nhưng sử dụng ánh xạ trong inputs
làm nguồn cấp dữ liệu.
Chạy
Status Run( const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs ) const
Tương tự như trên. Ngoài ra còn chạy các hoạt động trong run_outputs
.
Chạy
Status Run( const RunOptions & run_options, const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs, RunMetadata *run_metadata ) const
Sử dụng run_options
để bật hồ sơ hiệu suất.
run_metadata
, nếu không phải là null, sẽ được điền vào các kết quả lược tả.
Chạy
Status Run( const RunOptions & run_options, const FeedType & inputs, const std::vector< Output > & fetch_outputs, const std::vector< Operation > & run_outputs, std::vector< Tensor > *outputs, RunMetadata *run_metadata, const thread::ThreadPoolOptions & threadpool_options ) const
Tương tự như trên.
Ngoài ra, cho phép người dùng cung cấp triển khai threadpool tùy chỉnh thông qua ThreadPoolOptions.
Chạy có thể gọi được
Status RunCallable( CallableHandle handle, const std::vector< Tensor > & feed_tensors, std::vector< Tensor > *fetch_tensors, RunMetadata *run_metadata )
Gọi đồ thị con được đặt tên bằng handle
với các tùy chọn đã cho và các tensor đầu vào.
Thứ tự của các tenxơ trong feed_tensors
phải khớp với thứ tự tên trong CallableOptions::feed()
và thứ tự các tenxơ trong fetch_tensors
sẽ khớp với thứ tự tên trong CallableOptions::fetch()
khi sơ đồ con này được tạo. LƯU Ý: API này vẫn đang thử nghiệm và có thể thay đổi.
Chạy có thể gọi được
Status RunCallable( CallableHandle handle, const std::vector< Tensor > & feed_tensors, std::vector< Tensor > *fetch_tensors, RunMetadata *run_metadata, const thread::ThreadPoolOptions & options )
Gọi đồ thị con được đặt tên bằng handle
với các tùy chọn đã cho và các tensor đầu vào.
Thứ tự của các tenxơ trong feed_tensors
phải khớp với thứ tự tên trong CallableOptions::feed()
và thứ tự các tenxơ trong fetch_tensors
sẽ khớp với thứ tự tên trong CallableOptions::fetch()
khi sơ đồ con này được tạo. LƯU Ý: API này vẫn đang thử nghiệm và có thể thay đổi.
~Phiên khách hàng
~ClientSession()