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.
Métodos públicos
estático <T estende Número, U, V estende Número> RaggedGather <T, U> | |
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 List< Output <T>> outputNestedSplits ()
Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.