AutoShardDataset

AutoShardDataset classe final pública

Cria um conjunto de dados que fragmenta o conjunto de dados de entrada.

Cria um conjunto de dados que fragmenta o conjunto de dados de entrada por num_workers, retornando um conjunto de dados fragmentado para o index-th worker. Isso tenta fragmentar automaticamente um conjunto de dados examinando o gráfico do conjunto de dados e inserindo um fragmento op antes das entradas para um conjunto de dados do leitor (por exemplo, CSVDataset, TFRecordDataset).

Este conjunto de dados gerará um erro NotFound se não pudermos fragmentar o conjunto de dados automaticamente.

Classes aninhadas

classe AutoShardDataset.Options Atributos opcionais para AutoShardDataset

Métodos Públicos

Output <Object>
asOutput ()
Retorna o identificador simbólico de um tensor.
estáticos AutoShardDataset.Options
autoShardPolicy (Long autoShardPolicy)
estática AutoShardDataset
criar ( Scope escopo, Operando <?> inputDataset, Operando <longo> numWorkers, Operando <longo> índice, List <Class <? >> outputTypes, List < Forma > outputShapes, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação AutoShardDataset.
Output <?>
pega ()
estáticos AutoShardDataset.Options
numReplicas (Long numReplicas)

Métodos herdados

Métodos Públicos

pública Output <Object> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static AutoShardDataset.Options autoShardPolicy (Long autoShardPolicy)

public static AutoShardDataset criar ( Scope escopo, Operando <?> inputDataset, Operando <longo> numWorkers, Operando <longo> índice, List <Class <? >> outputTypes, List < Forma > outputShapes, Options ... Opções)

Método de fábrica para criar uma classe que envolve uma nova operação AutoShardDataset.

Parâmetros
alcance escopo atual
inputDataset Um tensor variante que representa o conjunto de dados de entrada.
numWorkers Um escalar que representa o número de trabalhadores para distribuir este conjunto de dados.
índice Um escalar que representa o índice do trabalhador atual de num_workers.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de AutoShardDataset

pública de saída <?> pega ()

public static AutoShardDataset.Options numReplicas (Long numReplicas)