SplitDedupData

classe final pública SplitDedupData

Uma operação divide a tupla XLA de dados de desduplicação de entrada em tensores inteiros e de ponto flutuante.

Os dados de desduplicação são uma tupla XLA, que consiste em valores inteiros e de ponto flutuante. Esta operação é dividir esses valores em dois grupos para dois tipos e construir cada grupo como um tensor para retornar.

Classes aninhadas

aula SplitDedupData.Options Atributos opcionais para SplitDedupData

Métodos públicos

SplitDedupData.Options estáticos
configuração (string de configuração)
static <T estende o número, U estende o número> SplitDedupData <T, U>
create ( Scope scope, Operand <?> input, Class<T> integerType, Class<U> floatType, String tupleMask, Options... options)
Método de fábrica para criar uma classe envolvendo uma nova operação SplitDedupData.
Saída <U>
floatTensor ()
Um tensor flutuante 1-D inclui elementos flutuantes da tupla de dados de desduplicação.
Saída <T>
inteiroTensor ()
Um tensor inteiro 1-D inclui elementos inteiros da tupla de dados de desduplicação.

Métodos Herdados

Métodos públicos

Configuração estática pública de SplitDedupData.Options (configuração de string)

public static SplitDedupData <T, U> create ( Scope scope, Operand <?> input, Class<T> integerType, Class<U> floatType, String tupleMask, Options... options)

Método de fábrica para criar uma classe envolvendo uma nova operação SplitDedupData.

Parâmetros
escopo escopo atual
entrada Uma tupla XLA incluindo elementos inteiros e flutuantes como tupla de dados de desduplicação.
integerType tipo integer_tensor. Tipos permitidos: int32, int64, uint32, uint64.
floatType tipo float_tensor. Tipos permitidos: half, bfloat16, float.
tupleMask Uma string TensorProto serializada da máscara de tupla de saída. Essa máscara é um tensor 2-D, com a primeira coluna como tipo de elemento tupla e a segunda coluna como span desse tipo. Por exemplo, uma tupla de saída de (1, 2, 0,1, 3), sua máscara é [[0, 2], [1, 1], [0, 1]]. Esperamos apenas dois tipos de elementos: integer(0) e float(1).
opções carrega valores de atributos opcionais
devoluções
  • uma nova instância de SplitDedupData

public Output <U> floatTensor ()

Um tensor flutuante 1-D inclui elementos flutuantes da tupla de dados de desduplicação.

Public Output <T> integerTensor ()

Um tensor inteiro 1-D inclui elementos inteiros da tupla de dados de desduplicação.