Crea un set di dati che suddivide in partizioni il set di dati di input.
Crea un set di dati che suddivide il set di dati di input in base a num_workers, restituendo un set di dati partizionato per il lavoratore indice-esimo. Questo tenta di partizionare automaticamente un set di dati esaminando il grafico del set di dati e inserendo un'operazione di shard prima degli input in un set di dati del lettore (ad esempio CSVDataset, TFRecordDataset).
Questo set di dati genererà un errore NotFound se non è possibile partizionare automaticamente il set di dati.
Classi nidificate
classe | AutoShardDataset.Options | Attributi facoltativi per AutoShardDataset |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita < TType > | comeuscita () Restituisce l'handle simbolico del tensore. |
AutoShardDataset.Options statico | autoShardPolicy (autoShardPolicy lunga) |
AutoShardDataset statico | |
Uscita <?> | maniglia () |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica < TType > asOutput ()
Restituisce l'handle simbolico del tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
creazione AutoShardDataset pubblica statica ( ambito ambito , Operando <?> inputDataset, Operando < TInt64 > numWorkers, Operando < TInt64 > indice, List<Class<? estende TType >> outputTypes, List< Shape > outputShapes, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione AutoShardDataset.
Parametri
scopo | ambito attuale |
---|---|
inputDataset | Un tensore variante che rappresenta il set di dati di input. |
numLavoratori | Uno scalare che rappresenta il numero di lavoratori su cui distribuire questo set di dati. |
indice | Uno scalare che rappresenta l'indice del lavoratore corrente su num_workers. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di AutoShardDataset