Reúna fatias irregulares do eixo `params`` 0` de acordo com os `índices`.
Saídas um `saída RaggedTensor` composto de output_dense_values`` e `output_nested_splits`, de tal modo 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.
- `índices` é um tensor denso com dtype` int32` ou `int64`, indicando quais valores devem ser reunidos.
- `output = ragged.from_nested_row_splits (output_dense_values, output_nested_splits)` é o tensor de saída.
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
estática <T estende TNumber , U estende TType > RaggedGather <T, U> | |
Output <U> | outputDenseValues () Os `flat_values` para o RaggedTensor retornado. |
List < Output <T >> | outputNestedSplits () Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado. |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
public static RaggedGather <T, U> create ( Scope escopo, Iterable < Operando <T >> paramsNestedSplits, Operando <u> paramsDenseValues, Operando <? estende TNumber > índices, Long OUTPUTRAGGEDRANK)
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 RaggedTensor` params`. |
paramsDenseValues | O `flat_values` para o` params` RaggedTensor. Houve uma mudança de terminologia no nível python de dense_values para flat_values, portanto, dense_values é o nome obsoleto. |
índices | Índices na dimensão mais externa dos `parâmetros` dos valores que devem ser reunidos. |
OUTPUTRAGGEDRANK | A classificação irregular do RaggedTensor de saída. `output_nested_splits` irá 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 de RaggedGather
Lista pública < saída <T >> outputNestedSplits ()
Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.