RaggedGather

RaggedGather classe final pública

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

Métodos Públicos

estática <T estende Número, U, V estende Number> RaggedGather <T, U>
criar ( Scope escopo, Iterable < Operando <T >> paramsNestedSplits, Operando <u> paramsDenseValues, Operando <V> índices, Longo OUTPUTRAGGEDRANK)
Método de fábrica para criar uma classe envolvendo uma nova operação RaggedGather.
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

Métodos Públicos

public static RaggedGather <T, U> create ( Scope escopo, Iterable < Operando <T >> paramsNestedSplits, Operando <u> paramsDenseValues, Operando <V> í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

pública Output <U> outputDenseValues ()

Os `flat_values` para o RaggedTensor retornado.

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

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