AutoShardDataset

AutoShardDataset de clase final pública

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 Atributos opcionales para AutoShardDataset

Métodos públicos

Salida <Objeto>
asOutput ()
Devuelve el identificador simbólico de un tensor.
AutoShardDataset.Options estático
AutoShardDataset estático
create ( ámbito de alcance, operando <?> inputDataset, operando <Long> numWorkers, operando <Long> índice, List <Class <? >> outputTypes, List < Shape > outputShapes, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación AutoShardDataset.
Salida <?>
AutoShardDataset.Options estático
numReplicas (Long numReplicas)

Métodos heredados

Métodos públicos

Salida pública <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.Options autoShardPolicy (Long autoShardPolicy)

public static AutoShardDataset create ( alcance de alcance, operando <?> inputDataset, operando <Long> numWorkers, operando <Long> índice, List <Class <? >> outputTypes, List < Shape > 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

Salida pública <?> handle ()

public static AutoShardDataset.Options numReplicas (Long numReplicas)