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]
- `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 o número, U, V estende o número> RaggedGather <T, U> | |
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 de 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
Lista pública< Saída <T>> saídaNestedSplits ()
Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.