Драйвер для выполнения Graph
.
Экземпляр Session
инкапсулирует среду, в которой выполняются Operation
в Graph
для вычисления Tensors
. Например:
// 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 . |
Публичные конструкторы
Сессия ( график g, конфигурация ConfigProto ) Создайте новый сеанс со связанным Graph и параметрами конфигурации. |
Публичные методы
пустота | закрывать () Освободите ресурсы, связанные с сеансом. |
пустота | восстановить (строковый префикс) Восстановите фактическое состояние переменных графика этого сеанса. |
пустота | |
пустота | запустить (String opName) Выполняет операцию в графе с заданным именем. |
пустота | запуститьИнит () Выполните инициализаторы графа. |
Сессия.Бегун | бегун () Создайте Runner для выполнения операций с графом и оценки тензоров. |
пустота | сохранить (префикс строки) Сохраняет фактическое состояние переменных графика этого сеанса. |
Унаследованные методы
Публичные конструкторы
публичная сессия ( график g)
публичный сеанс ( график g, конфигурация ConfigProto )
Создайте новый сеанс со связанным Graph
и параметрами конфигурации.
Параметры
г | Graph на котором будет работать созданный сеанс. |
---|---|
конфигурация | Параметры конфигурации для сеанса, заданные как буфер протокола ConfigProto . |
Броски
IllegalArgumentException | если конфигурация не является допустимой сериализацией буфера протокола ConfigProto. |
---|
Публичные методы
public void закрыть ()
Освободите ресурсы, связанные с сеансом.
Блокируется до тех пор, пока не прекратятся активные выполнения (вызовы run()
). Сеанс нельзя использовать после закрытия.
общедоступное восстановление void (префикс String)
Восстановите фактическое состояние переменных графика этого сеанса.
prefix
— это путь, по которому находятся файлы, содержащие состояние переменных, за которым следует префикс имени файла. Например, если для prefix
установлено значение mymodel/myvariables/variables , файлы загружаются из mymodel/myvariables и называютсяvariables.data -*-of-*.
Обратите внимание, что этот метод может изменить базовый график, если один из его сеансов сохраняется впервые. Дополнительные сведения см. в ERROR(/Graph#saverDef())
.
Параметры
префикс | префикс для восстановления из |
---|
публичный забег в пустоту ( оп оп)
Выполняет операцию в графе.
Этот метод эквивалентен session.runner().addTarget(op).run()
.
Параметры
оп | операцию для запуска. |
---|
public void run (String opName)
Выполняет операцию в графе с заданным именем.
Этот метод эквивалентен session.runner().addTarget(opName).run()
.
Параметры
имя_оператора | имя операции, которую нужно выполнить. |
---|
Броски
IllegalArgumentException | если в графе не найдена операция с таким именем |
---|
общественный недействительный runInit ()
Выполните инициализаторы графа.
Этот метод эквивалентен session.run(Ops.create(session.graph).init())
.
общедоступный бегун Session.Runner ()
Создайте Runner для выполнения операций с графом и оценки тензоров.
публичное сохранение пустоты (префикс строки)
Сохраняет фактическое состояние переменных графика этого сеанса.
prefix
— это путь, по которому будут сохранены файлы, содержащие состояние переменных, за которым следует префикс для именования этих файлов. Например, если для prefix
установлено значение mymodel/myvariables/variables , то сгенерированные файлы будут расположены в папке mymodel/myvariables и названы переменными.data-*-of-*.
Обратите внимание, что этот метод может изменить базовый график, если один из его сеансов сохраняется впервые. Дополнительные сведения см. в ERROR(/Graph#saverDef())
.
Параметры
префикс | префикс переменных файлов для сохранения |
---|