RaggedGather

classe final pública RaggedGather

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

Produz 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.
(Observação: esta operação c++ é usada para implementar a operação python `tf.ragged.gather` de nível superior, que também oferece suporte a índices irregulares.)

Métodos Públicos

estático <T estende o número, U, V estende o número> RaggedGather <T, U>
create ( Escopo , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> índices, Long OUTPUTRAGGEDRANK)
Método de fábrica para criar uma classe que envolve uma nova operação RaggedGather.
Saída <U>
saídaDenseValues ()
Os `flat_values` para o RaggedTensor retornado.
Lista< Saída <T>>
saídaNestedSplits ()
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 escopo, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> índices, Long OUTPUTRAGGEDRANK)

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

Parâmetros
escopo escopo atual
parâmetrosNestedSplits Os tensores `nested_row_splits` que definem o particionamento de linha para a entrada `params` RaggedTensor.
paramsDenseValues Os `flat_values` para o RaggedTensor `params`. Houve uma mudança de terminologia no nível python de Dense_values ​​para Flat_values, então Dense_values ​​é o nome obsoleto.
índices Índices na dimensão mais externa dos `parâmetros` dos valores que devem ser coletados.
OUTPUTRAGGEDRANK 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

Saída pública <U> saídaDenseValues ()

Os `flat_values` para o RaggedTensor retornado.

Lista pública< Saída <T>> saídaNestedSplits ()

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