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
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. |
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. |
Phương pháp kế thừa
Nhà xây dựng công cộng
Phiên công khai ( Cấu hình g, byte[])
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 | Các tham số cấu hình cho phiên được chỉ định làm vùng đệm giao thức ConfigProto được tuần tự hóa. |
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.
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.