RaggedGather

RaggedGather public final class

Reúna cortes irregulares del eje "params" "0" de acuerdo con los "índices".

Salidas de un `salida RaggedTensor` compuesta a partir de` `output_dense_values` y 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]
 
donde
  • `params = ragged.from_nested_row_splits (params_dense_values, params_nested_splits)` proporciona los valores que deben recopilarse.
  • `índices` es un tensor denso con dtype` int32` o `int64`, que indica qué valores deben recopilarse.
  • `output = ragged.from_nested_row_splits (output_dense_values, output_nested_splits)` es el tensor de salida.
(Nota: esta operación de C ++ se usa para implementar la operación de python de nivel superior `tf.ragged.gather`, que también admite índices irregulares).

Métodos públicos

estática <T se extiende Número, U, V se extiende Número> RaggedGather <T, U>
crear ( Alcance alcance, Iterable < Operando <T >> paramsNestedSplits, operando <u> paramsDenseValues, operando <V> índices, a largo OUTPUTRAGGEDRANK)
Método de fábrica para crear una clase que envuelva una nueva operación RaggedGather.
Salida <U>
outputDenseValues ()
Los `flat_values` para el RaggedTensor devuelto.
Lista < salida <T >>
outputNestedSplits ()
Los tensores `nested_row_splits` que definen la división de filas para el RaggedTensor devuelto.

Métodos heredados

Métodos públicos

public static RaggedGather <T, U> crear ( Alcance alcance, Iterable < operando <T >> paramsNestedSplits, operando <u> paramsDenseValues, operando <v> índices, Long OUTPUTRAGGEDRANK)

Método de fábrica para crear una clase que envuelva una nueva operación RaggedGather.

Parámetros
alcance alcance actual
paramsNestedSplits Los tensores `nested_row_splits` que definen la división de filas para la entrada de RaggedTensor` params`.
paramsDenseValues Los `flat_values` para el` params` RaggedTensor. Hubo un cambio de terminología a nivel de Python de dense_values ​​a flat_values, por lo que dense_values ​​es el nombre obsoleto.
índices Índices en la dimensión más externa de "params" de los valores que deben recopilarse.
OUTPUTRAGGEDRANK El rango desigual de la salida RaggedTensor. `output_nested_splits` contendrá este número de tensores` row_splits`. Este valor debe ser igual a `indices.shape.ndims + params.ragged_rank - 1`.
Devoluciones
  • una nueva instancia de RaggedGather

pública de salida <U> outputDenseValues ()

Los `flat_values` para el RaggedTensor devuelto.

Lista pública < salida <T >> outputNestedSplits ()

Los tensores `nested_row_splits` que definen la división de filas para el RaggedTensor devuelto.