Crea un set di dati che divide il set di dati di input.
Crea un set di dati che divide il set di dati di input da num_workers, restituendo un set di dati partizionato per il worker index -esimo. Questo tenta di suddividere automaticamente un set di dati esaminando il grafico del set di dati e inserendo un'operazione di frammentazione prima degli input in un set di dati del lettore (ad esempio CSVDataset, TFRecordDataset).
Questo set di dati genererà un errore NotFound se non possiamo suddividere automaticamente il set di dati.
Classi annidate
classe | ExperimentalAutoShardDataset.Options | Attributi facoltativi per ExperimentalAutoShardDataset |
Metodi pubblici
Output <Object> | asOutput () Restituisce la maniglia simbolica di un tensore. |
static ExperimentalAutoShardDataset.Options | autoShardPolicy (Long autoShardPolicy) |
static ExperimentalAutoShardDataset | |
Uscita <?> | maniglia () |
Metodi ereditati
Metodi pubblici
output pubblico <Object> asOutput ()
Restituisce la maniglia simbolica di un 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.
public static ExperimentalAutoShardDataset create ( Scope scope, Operand <?> inputDataset, Operand <Long> numWorkers, Operand <Long> index, List <Class <? >> outputTypes, List < Shape > outputShapes, Options ... options)
Metodo Factory per creare una classe che racchiude una nuova operazione ExperimentalAutoShardDataset.
Parametri
scopo | ambito attuale |
---|---|
inputDataset | Un tensore variante che rappresenta il set di dati di input. |
numWorkers | Uno scalare che rappresenta il numero di worker 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 ExperimentalAutoShardDataset