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.
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático <T estende TNumber , U estende TType > 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
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
public static RaggedGather <T, U> create ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? estende TNumber > í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.