RaggedGather

public final class RaggedGather

Reúna fatias irregulares do eixo `params`` 0` de acordo com os `índices`.

Produz uma saída `RaggedTensor` composta de` output_dense_values` e `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]
 

onde

• `params = ragged.from_nested_row_splits (params_dense_values, params_nested_splits)` fornece os valores que devem ser coletados.
• `índices` é um tensor denso com dtype` int32` ou `int64`, indicando quais valores devem ser reunidos.
• `output = ragged.from_nested_row_splits (output_dense_values, output_nested_splits)` é o tensor de saída.

(Observação: este c ++ op é usado para implementar o op python de nível superior `tf.ragged.gather`, que também suporta índices irregulares.

Métodos Públicos

static <T extends Number, U, V extends Number> RaggedGather <T, U>	criar ( Scope escopo, Iterable < Operando <T >> paramsNestedSplits, Operando <u> paramsDenseValues, Operando <V> índices, Longo OUTPUTRAGGEDRANK) Método de fábrica para criar uma classe envolvendo uma nova operação RaggedGather.
Saída <U>	outputDenseValues () Os `flat_values` para o RaggedTensor retornado.
Listar < Output <T>>	outputNestedSplits () Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

final booleano	igual a (objeto obj)
final int	hashCode ()
Operação	op () Retorna a Operation subjacente
corda final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notificar tudo ()
Fragmento	toString ()
vazio final	espera (long arg0, int arg1)
vazio final	espera (long arg0)
vazio final	espera ()

Métodos Públicos

public static RaggedGather <T, U> create ( Scope escopo, Iterable < Operando <T >> paramsNestedSplits, Operando <u> paramsDenseValues, Operando <V> índices, Long OUTPUTRAGGEDRANK)

Método de fábrica para criar uma classe envolvendo uma nova operação RaggedGather.

Parâmetros

alcance	escopo atual
paramsNestedSplits	Os tensores `nested_row_splits` que definem o particionamento de linha para a entrada RaggedTensor` params`.
paramsDenseValues	O `flat_values` para o` params` RaggedTensor. Houve uma mudança de terminologia no nível python de dense_values ​​para flat_values, portanto, dense_values ​​é o nome obsoleto.
índices	Índices na dimensão mais externa de `params` dos valores que devem ser reunidos.
OUTPUTRAGGEDRANK	A classificação irregular do RaggedTensor de saída. `output_nested_splits` irá conter este número de tensores` row_splits`. Este valor deve ser igual a `indices.shape.ndims + params.ragged_rank - 1`.

Devoluções

• uma nova instância de RaggedGather

public Output <U> outputDenseValues ()

Os `flat_values` para o RaggedTensor retornado.

public List < Output <T>> outputNestedSplits ()

Os tensores `nested_row_splits` que definem o particionamento de linha para o RaggedTensor retornado.