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.
Métodos públicos
estático <T extiende Número, U, V extiende Número> RaggedGather <T, U> | |
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
lista pública< Salida <T>> salidaNestedSplits ()
Los tensores `nested_row_splits` que definen la partición de filas para el RaggedTensor devuelto.