RaggedGather

clase final pública RaggedGather

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

Genera una salida `RaggedTensor` compuesta por `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 se deben recopilar.
  • `indices` 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 utiliza para implementar la operación de Python de nivel superior `tf.ragged.gather`, que también admite índices irregulares).

Constantes

Cadena OP_NOMBRE El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Métodos públicos

estático <T extiende TNumber , U extiende TType > RaggedGather <T, U>
crear ( alcance alcance, Iterable< Operando <T>> paramsNestedSplits, Operando <U> paramsDenseValues, Operando <? extiende TNumber > índices, Long OUTPUTRAGGEDRANK)
Método de fábrica para crear una clase que envuelve una nueva operación de RaggedGather.
Salida <U>
salidaValoresDensos ()
Los `flat_values` para el RaggedTensor devuelto.
Lista< Salida <T>>
salidaNestedSplits ()
Los tensores `nested_row_splits` que definen la partición de filas para el RaggedTensor devuelto.

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "RaggedGather"

Métodos públicos

público estático RaggedGather <T, U> crear ( alcance alcance , Iterable< Operando <T>> paramsNestedSplits, Operando <U> paramsDenseValues, Operando <? extiende TNumber > índices, Long OUTPUTRAGGEDRANK)

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

Parámetros
alcance alcance actual
paramsNestedSplits Los tensores `nested_row_splits` que definen la partición de filas para la entrada RaggedTensor `params`.
paramsDenseValues Los `flat_values` para el `params` RaggedTensor. Hubo un cambio de terminología en el nivel de Python de valores_densos a valores_planos, por lo que valores_densos es el nombre obsoleto.
índices Índices en la dimensión más externa de los "parámetros" de los valores que deben recopilarse.
RANGO DE SALIDA El rango irregular del RaggedTensor de salida. `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

Salida pública <U> salidaDenseValues ​​()

Los `flat_values` para el RaggedTensor devuelto.

Lista pública< Salida <T>> salidaNestedSplits ()

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