Graph
실행용 드라이버입니다.
Session
인스턴스는 Tensors
계산하기 위해 Graph
의 Operation
이 실행되는 환경을 캡슐화합니다. 예를 들어:
// 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
}
}
}
경고: Session
close()
호출하여 명시적으로 해제 해야 하는 리소스를 소유합니다.
세션 인스턴스는 스레드로부터 안전합니다.
중첩 클래스
수업 | 세션.실행 | 세션을 실행할 때 얻은 출력 텐서 및 메타데이터입니다. | |
수업 | 세션.러너 | Operation 을 실행하고 Tensors 평가합니다. |
공공 생성자
공개 방법
무효의 | 닫다 () 세션과 관련된 리소스를 해제합니다. |
무효의 | 복원 (문자열 접두사) 이 세션 그래프 변수의 실제 상태를 복원합니다. |
무효의 | |
무효의 | 실행 (문자열 opName) 주어진 이름으로 그래프에서 작업을 실행합니다. |
무효의 | 실행 초기화 () 그래프의 초기화 프로그램을 실행합니다. |
세션.러너 | 주자 () 그래프 작업을 실행하고 Tensor를 평가하는 Runner를 만듭니다. |
무효의 | 저장 (문자열 접두사) 이 세션 그래프의 변수 실제 상태를 저장합니다. |
상속된 메서드
공공 생성자
공개 세션 ( 그래프 g, ConfigProto 구성)
관련 Graph
및 구성 옵션을 사용하여 새 세션을 구성합니다.
매개변수
g | 생성된 세션이 작동할 Graph 입니다. |
---|---|
구성 | ConfigProto 프로토콜 버퍼로 지정된 세션에 대한 구성 매개변수입니다. |
던지기
IllegalArgumentException | 구성이 ConfigProto 프로토콜 버퍼의 유효한 직렬화가 아닌 경우. |
---|
공개 방법
공개 무효 복원 (문자열 접두사)
이 세션 그래프 변수의 실제 상태를 복원합니다.
prefix
는 변수 상태를 포함하는 파일이 있는 경로이고 그 뒤에 파일 이름 접두사가 옵니다. 예를 들어 prefix
mymodel/myvariables/variables 로 설정된 경우 파일은 mymodel/myvariables 에서 로드되고 이름은 Variable.data-*-of-* 로 지정됩니다.
이 메서드는 세션 중 하나가 처음으로 저장되는 경우 기본 그래프를 변경할 수 있습니다. 자세한 내용은 ERROR(/Graph#saverDef())
참조하세요.
매개변수
접두사 | 복원할 접두사 |
---|
공개 무효 실행 ( Op op)
그래프에서 작업을 실행합니다.
이 메서드는 session.runner().addTarget(op).run()
과 동일합니다.
매개변수
작전 | 실행할 작업. |
---|
공개 무효 실행 (문자열 opName)
주어진 이름으로 그래프에서 작업을 실행합니다.
이 메서드는 session.runner().addTarget(opName).run()
과 동일합니다.
매개변수
opName | 실행할 작업의 이름입니다. |
---|
던지기
IllegalArgumentException | 그래프에서 해당 이름의 작업을 찾을 수 없는 경우 |
---|
공공 무효 runInit ()
그래프의 초기화 프로그램을 실행합니다.
이 메서드는 session.run(Ops.create(session.graph).init())
와 동일합니다.
공개 무효 저장 (문자열 접두사)
이 세션 그래프의 변수 실제 상태를 저장합니다.
prefix
는 변수 상태를 포함하는 파일이 저장될 경로이고, 그 뒤에는 이러한 파일의 이름을 지정하기 위한 접두사가 옵니다. 예를 들어 prefix
mymodel/myvariables/variables 로 설정된 경우 생성된 파일은 mymodel/myvariables 아래에 위치하며 이름은 Variable.data-*-of-*로 지정됩니다.
이 메서드는 세션 중 하나가 처음으로 저장되는 경우 기본 그래프를 변경할 수 있습니다. 자세한 내용은 ERROR(/Graph#saverDef())
참조하세요.
매개변수
접두사 | 저장할 변수 파일의 접두어 |
---|