Crea un conjunto de datos que fragmenta el conjunto de datos de entrada.
Crea un conjunto de datos que fragmenta el conjunto de datos de entrada por num_workers, devolviendo un conjunto de datos fragmentado para el trabajador indexado. Esto intenta fragmentar automáticamente un conjunto de datos examinando el gráfico del conjunto de datos e insertando una operación de fragmentación antes de las entradas a un conjunto de datos del lector (por ejemplo, CSVDataset, TFRecordDataset).
Este conjunto de datos arrojará un error NotFound si no podemos fragmentar el conjunto de datos automáticamente.
Clases anidadas
clase | AutoShardDataset.Options | Los atributos opcionales para AutoShardDataset |
Métodos públicos
Salida <objeto> | asOutput () Devuelve el identificador simbólico de un tensor. |
estáticas AutoShardDataset.Options | autoShardPolicy (Long autoShardPolicy) |
estática AutoShardDataset | crear ( Alcance alcance, operando <?> inputDataset, operando <pulsación larga> numWorkers, operando <pulsación larga> <índice, Lista de marcas <? >> outputTypes, List < Forma > outputShapes, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación AutoShardDataset. |
Salida <?> | asa () |
estáticas AutoShardDataset.Options | numReplicas (Long numReplicas) |
Métodos heredados
Métodos públicos
pública de salida <Objeto> asOutput ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static AutoShardDataset crear ( Alcance alcance, operando <?> inputDataset, operando <pulsación larga> numWorkers, operando <pulsación larga> <índice, Lista de marcas <? >> outputTypes, List < Forma > outputShapes, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación AutoShardDataset.
Parámetros
alcance | alcance actual |
---|---|
inputDataset | Un tensor variante que representa el conjunto de datos de entrada. |
numWorkers | Un escalar que representa la cantidad de trabajadores para distribuir este conjunto de datos. |
índice | Un escalar que representa el índice del trabajador actual de num_workers. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de AutoShardDataset