RaggedGather

classe final pública RaggedGather

Reúna fatias irregulares do eixo `0` de `params` de acordo com `índices`.

Gera uma saída `RaggedTensor` composta de `output_dense_values` e `output_nested_splits`, tal que:

output.shape = indices.shape + params.shape[1:]
 output.ragged_rank = indices.shape.ndims + params.ragged_rank
 output[i...j, d0...dn] = params[indices[i...j], d0...dn]
 
onde
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` fornece os valores que devem ser coletados.
  • `indices` é um tensor denso com dtype `int32` ou `int64`, indicando quais valores devem ser coletados.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` é o tensor de saída.
(Nota: Este op c++ ​​é usado para implementar o op python `tf.ragged.gather` de nível superior, que também suporta índices ragged.)

Métodos públicos

estático <T estende Número, U, V estende Número> RaggedGather <T, U>
create ( Escopo do escopo, Iterable< Operando <T>> paramsNestedSplits, Operando <U> paramsDenseValues, Operando <V> índices, OUTPUTRAGGEDRANK longo)
Método de fábrica para criar uma classe envolvendo uma nova operação RaggedGather.
Saída <U>
outputDenseValues ()
Os `flat_values` para o RaggedTensor retornado.
Lista< Saída <T>>
outputNestedSplits ()
Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.

Métodos Herdados

Métodos públicos

public static RaggedGather <T, U> create ( Escopo do escopo, Iterable< Operando <T>> paramsNestedSplits, Operando <U> paramsDenseValues, Operando <V> índices, OUTPUTRAGGEDRANK longo)

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

Parâmetros
alcance escopo atual
paramsNestedSplits Os tensores `nested_row_splits` que definem o particionamento de linha para a entrada `params` do RaggedTensor.
paramsDenseValues Os `flat_values` para o `params` RaggedTensor. Houve uma mudança de terminologia no nível do python de valor_de_densidade para valor_de_plano, então valor_de_densidade é o nome obsoleto.
índices Índices na dimensão mais externa de `params` dos valores que devem ser coletados.
CLASSIFICAÇÃO DE SAÍDA A classificação irregular da saída RaggedTensor. `output_nested_splits` conterá este número de tensores `row_splits`. Este valor deve ser igual a `indices.shape.ndims + params.ragged_rank - 1`.
Devoluções
  • uma nova instância do RaggedGather

Public Output <U> outputDenseValues ()

Os `flat_values` para o RaggedTensor retornado.

public List< Output <T>> outputNestedSplits ()

Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.