Reúna cortes irregulares del eje `0` de `params` según 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]
- `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.
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> | |
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.
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 a 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. |
RENDIMIENTO 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
Lista pública< Salida <T>> salidaNestedSplits ()
Los tensores `nested_row_splits` que definen la partición de filas para el RaggedTensor devuelto.