Reúna cortes irregulares del eje "params" "0" de acuerdo con los "índices".
Salidas de un `salida RaggedTensor` compuesta a partir de` `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 deben recopilarse.
- `índices` 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
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estática <T se extiende TNumber , U se extiende Ttype > RaggedGather <T, U> | |
Salida <U> | outputDenseValues () Los `flat_values` para el RaggedTensor devuelto. |
Lista < salida <T >> | outputNestedSplits () Los tensores `nested_row_splits` que definen la división de filas para el RaggedTensor devuelto. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
public static 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 envuelva una nueva operación RaggedGather.
Parámetros
alcance | alcance actual |
---|---|
paramsNestedSplits | Los tensores `nested_row_splits` que definen la división de filas para la entrada de RaggedTensor` params`. |
paramsDenseValues | Los `flat_values` para el` params` RaggedTensor. Hubo un cambio de terminología a nivel de Python de dense_values a flat_values, por lo que dense_values es el nombre obsoleto. |
índices | Índices en la dimensión más externa de "params" de los valores que deben recopilarse. |
OUTPUTRAGGEDRANK | El rango desigual 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 >> outputNestedSplits ()
Los tensores `nested_row_splits` que definen la división de filas para el RaggedTensor devuelto.