Trình điều khiển để thực thi Graph
.
Một phiên bản Session
đóng gói môi trường trong đó Operation
trong Graph
được thực thi để tính toán Tensors
. Ví dụ:
// Let's say graph is an instance of the Graph class
// for the computation y = 3 * x
try (Session s = new Session(graph)) {
try (Tensor x = Tensor.create(2.0f);
Tensor y = s.runner().feed("x", x).fetch("y").run().get(0)) {
System.out.println(y.floatValue()); // Will print 6.0f
try (Tensor x = Tensor.create(1.1f);
Tensor y = s.runner().feed("x", x).fetch("y").run().get(0)) {
System.out.println(y.floatValue()); // Will print 3.3f
}
}
}
CẢNH BÁO: Session
sở hữu tài nguyên phải được giải phóng rõ ràng bằng cách gọi close()
.
Các phiên bản của Phiên là an toàn theo luồng.
Các lớp lồng nhau
lớp học | Phiên.Run | Các tensor đầu ra và siêu dữ liệu thu được khi thực hiện một phiên. | |
lớp học | Phiên.Runner | Chạy Operation s và đánh giá Tensors . |
Nhà xây dựng công cộng
Phiên ( Biểu đồ g, cấu hình ConfigProto ) Xây dựng một phiên mới với các tùy chọn Graph và biểu đồ liên quan. |
Phương pháp công khai
trống rỗng | đóng () Phát hành tài nguyên được liên kết với Phiên. |
trống rỗng | khôi phục (Tiền tố chuỗi) Khôi phục trạng thái thực tế của các biến trong biểu đồ của phiên này. |
trống rỗng | |
trống rỗng | chạy (Chuỗi opName) Thực hiện một thao tác trong biểu đồ với tên đã cho. |
trống rỗng | runInit () Thực hiện các công cụ khởi tạo của biểu đồ. |
Phiên.Runner | người chạy () Tạo một Runner để thực hiện các thao tác trên đồ thị và đánh giá Tensors. |
trống rỗng | lưu (Tiền tố chuỗi) Lưu trạng thái thực tế của các biến trong biểu đồ của phiên này. |
Phương pháp kế thừa
Nhà xây dựng công cộng
Phiên công khai ( Đồ thị g)
Phiên công khai ( Đồ thị g, cấu hình ConfigProto )
Xây dựng một phiên mới với các tùy chọn Graph
và biểu đồ liên quan.
Thông số
g | Graph mà Phiên đã tạo sẽ hoạt động trên đó. |
---|---|
cấu hình | Tham số cấu hình cho phiên được chỉ định làm vùng đệm giao thức ConfigProto . |
Ném
Đối số bất hợp phápNgoại lệ | nếu cấu hình không phải là sự tuần tự hóa hợp lệ của bộ đệm giao thức ConfigProto. |
---|
Phương pháp công khai
đóng khoảng trống công khai ()
Phát hành tài nguyên được liên kết với Phiên.
Chặn cho đến khi không còn hoạt động thực thi nào (các lệnh gọi run()
). Phiên không thể sử dụng được sau khi đóng lại.
khôi phục khoảng trống công khai (Tiền tố chuỗi)
Khôi phục trạng thái thực tế của các biến trong biểu đồ của phiên này.
prefix
là đường dẫn chứa các tệp chứa trạng thái biến, theo sau là tiền tố tên tệp. Ví dụ: nếu prefix
được đặt thành mymodel/myvariables/variables thì các tệp sẽ được tải từ mymodel/myvariables và được đặt tên là biến.data-*-of-*
Lưu ý rằng phương pháp này có thể thay đổi biểu đồ cơ bản nếu đây là lần đầu tiên một trong các phiên của nó được lưu, hãy xem ERROR(/Graph#saverDef())
để biết thêm chi tiết.
Thông số
tiền tố | tiền tố để khôi phục từ |
---|
chạy void công khai ( Op op)
Thực hiện một thao tác trong biểu đồ.
Phương thức này tương đương với session.runner().addTarget(op).run()
.
Thông số
op | thao tác cần chạy. |
---|
chạy void công khai (Chuỗi opName)
Thực hiện một thao tác trong biểu đồ với tên đã cho.
Phương thức này tương đương với session.runner().addTarget(opName).run()
.
Thông số
tên op | tên của hoạt động để chạy. |
---|
Ném
Đối số bất hợp phápNgoại lệ | nếu không tìm thấy hoạt động nào của tên đó trong biểu đồ |
---|
public void runInit ()
Thực hiện các công cụ khởi tạo của biểu đồ.
Phương thức này tương đương với session.run(Ops.create(session.graph).init())
.
phiên công khai.Runner Á hậu ( )
Tạo một Runner để thực hiện các thao tác trên đồ thị và đánh giá Tensors.
lưu void công khai (Tiền tố chuỗi)
Lưu trạng thái thực tế của các biến trong biểu đồ của phiên này.
prefix
là đường dẫn nơi các tệp chứa trạng thái biến sẽ được lưu, theo sau là tiền tố để đặt tên cho các tệp này. Ví dụ: nếu prefix
được đặt thành mymodel/myvariables/variables thì các tệp được tạo sẽ nằm trong mymodel/myvariables và được đặt tên là biến.data-*-of-*
Lưu ý rằng phương pháp này có thể thay đổi biểu đồ cơ bản nếu đây là lần đầu tiên một trong các phiên của nó được lưu, hãy xem ERROR(/Graph#saverDef())
để biết thêm chi tiết.
Thông số
tiền tố | tiền tố vào các tập tin biến để lưu |
---|