RaggedGather

clase final pública RaggedGather

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

Genera una salida `RaggedTensor` compuesta por `output_dense_values` y `output_nested_splits`, de 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]
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 se deben recopilar.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` es el tensor de salida.
(Nota: esta operación c++ se usa para implementar la operación python `tf.ragged.gather` de nivel superior, 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 ( Á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 RaggedGather.
Salida <U>
valores densos de salida ()
Los `flat_values` para el RaggedTensor devuelto.
Lista< Salida <T>>
Salida de divisiones anidadas ()
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 ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> indices, Long OUTPUTRAGGEDRANK)

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

Parámetros
alcance alcance actual
paramsNestedSplits Los tensores `nested_row_splits` que definen la partición de filas para la entrada `params` RaggedTensor.
paramsDenseValues Los `flat_values` para `params` RaggedTensor. Hubo un cambio de terminología en el nivel de python de dense_values ​​a flat_values, por lo que dense_values ​​es el nombre en desuso.
índices Índices en la dimensión más externa de `params` de los valores que se deben recopilar.
OUTPUTRAGGEDRANK El rango irregular 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

Salida pública <U> valores densos de salida ()

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.