SavedModelBundle rappresenta un modello caricato dall'archivio.
Il modello consiste in una descrizione del calcolo (un Graph ), una Session con tensori (ad esempio, parametri o variabili nel grafico) inizializzati su valori salvati in memoria e una descrizione del modello come buffer del protocollo MetaGraphDef .
Classi nidificate
| classe | SavedModelBundle.Exporter | Opzioni per esportare un modello salvato. | |
| classe | SavedModelBundle.Loader | Opzioni per caricare un SavedModel. | |
Costanti
| Corda | DEFAULT_TAG |
Metodi pubblici
| Mappa<Stringa, Tensore > | |
| vuoto | |
| SavedModelBundle.Exporter statico | esportatore (Stringa exportDir) Esporta un modello salvato. |
| FunzioneConcreta | funzione (Stringa SignatureKey) Restituisce una ConcreteFunction corrispondente alla firma della funzione. |
| Grafico | grafico () Restituisce il grafico che descrive il calcolo eseguito dal modello. |
| SavedModelBundle statico | caricamento (stringa exportDir, stringa... tag) Carica un modello salvato da una directory di esportazione. |
| SavedModelBundle.Loader statico | caricatore (Stringa exportDir) Carica un modello salvato. |
| MetaGraphDef | metaGraphDef () Restituisce il buffer del protocollo MetaGraphDef associato al modello salvato. |
| Sessione | |
| Elenco <Firma> | firme () Restituisce la firma di tutte le funzioni disponibili in questo modello salvato. |
Metodi ereditati
Costanti
Stringa finale statica pubblica DEFAULT_TAG
Metodi pubblici
chiamata pubblica Map<String, Tensor > (argomenti Map<String, Tensor >)
Richiama la funzione predefinita direttamente da questo modello.
La selezione della funzione predefinita viene effettuata in base alla prima delle seguenti condizioni che è vera:
- La funzione è l'unica firma disponibile allegata al grafico principale di questo modello salvato
- La funzione è mappata sul nome della firma predefinita, che è "serving_default"
Il chiamante è responsabile della chiusura di tutti i Tensor restituiti.
Parametri
| argomenti | elenco di tensori di input, mappati in base al nome della firma |
|---|
ritorna
- elenco dei tensori di output, mappati in base al nome della firma
Lancia
| IllegalArgumentException | se nessuna funzione può essere selezionata per impostazione predefinita |
|---|
pubblico vuoto chiudi ()
esportatore pubblico statico SavedModelBundle.Exporter (String exportDir)
Esporta un modello salvato.
Restituisce un oggetto Exporter per impostare le opzioni di configurazione prima di salvare effettivamente il modello.
Parametri
| diresportazione | il percorso della directory contenente un modello salvato. |
|---|
funzione pubblica ConcreteFunction (String SignatureKey)
Restituisce una ConcreteFunction corrispondente alla firma della funzione.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
Parametri
| firmaKey | nome della SignatureDef nel modello salvato. |
|---|
ritorna
- oggetto che può essere utilizzato per effettuare chiamate a una funzione
Lancia
| IllegalArgumentException | se signatureKey non viene trovata in questo modello salvato. |
|---|
caricamento pubblico statico di SavedModelBundle (tag String exportDir, String...)
Carica un modello salvato da una directory di esportazione. Il modello che viene caricato deve essere creato utilizzando l' API del modello salvato .
Questo metodo è una scorciatoia per:
SavedModelBundle.loader().withTags(tags).load();
Parametri
| diresportazione | il percorso della directory contenente un modello salvato. |
|---|---|
| tag | i tag che identificano la metagraphdef specifica da caricare. |
ritorna
- un bundle contenente il grafico e la sessione associata.
caricatore pubblico statico SavedModelBundle.Loader (stringa exportDir)
Carica un modello salvato.
Restituisce un oggetto Loader che può impostare le opzioni di configurazione prima di caricare effettivamente il modello,
Parametri
| diresportazione | il percorso della directory contenente un modello salvato. |
|---|
metaGraphDef pubblico metaGraphDef ()
Restituisce il buffer del protocollo MetaGraphDef associato al modello salvato.
sessione pubblica ()
Restituisce la Session con cui eseguire il calcolo utilizzando il modello.
ritorna
- la sessione inizializzata
Elenco pubblico< Firma > firme ()
Restituisce la firma di tutte le funzioni disponibili in questo modello salvato.