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).

Métodos públicos

estático <T extiende Número, U, V extiende Número> 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 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

Métodos públicos

public static RaggedGather <T, U> create (Ámbito de alcance , Iterable< Operando <T>> paramsNestedSplits, Operando <U> paramsDenseValues, Operando <V> í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.