tensoreflusso:: ops:: SparseConcat

#include <sparse_ops.h>

Concatena un elenco di SparseTensor lungo la dimensione specificata.

Riepilogo

La concatenazione riguarda le versioni dense di questi tensori sparsi. Si presuppone che ogni input sia uno SparseTensor i cui elementi sono ordinati secondo un numero di dimensione crescente.

Tutte le forme degli input devono corrispondere, ad eccezione della dimensione concat. Gli elenchi indices , values e shapes devono avere la stessa lunghezza.

La forma dell'output è identica a quella degli input, tranne che lungo la dimensione concat, dove è la somma delle dimensioni degli input lungo quella dimensione.

Gli elementi di output verranno utilizzati per preservare l'ordinamento lungo il numero di dimensione crescente.

Questa operazione viene eseguita nel tempo O(M log M) , dove M è il numero totale di valori non vuoti in tutti gli input. Ciò è dovuto alla necessità di un ordinamento interno per concatenare in modo efficiente attraverso una dimensione arbitraria.

Ad esempio, se concat_dim = 1 e gli input sono

sp_inputs[0]: shape = [2, 3]
[0, 2]: "a"
[1, 0]: "b"
[1, 1]: "c"

sp_inputs[1]: shape = [2, 4]
[0, 1]: "d"
[0, 2]: "e"

quindi l'output sarà

shape = [2, 7]
[0, 2]: "a"
[0, 4]: "d"
[0, 5]: "e"
[1, 0]: "b"
[1, 1]: "c"

Graficamente questo equivale a fare

[    a] concat [  d e  ] = [    a   d e  ]
[b c  ]        [       ]   [b c          ]

Argomenti:

  • scope: un oggetto Scope
  • indici: 2-D. Indici di ogni ingresso SparseTensor .
  • valori: 1-D. Valori non vuoti di ogni SparseTensor .
  • forme: 1-D. Forme di ogni SparseTensor .
  • concat_dim: dimensione su cui concatenare. Deve essere compreso nell'intervallo [-rank, rango), dove rango è il numero di dimensioni in ciascun input SparseTensor .

Ritorna:

  • Output output_indices: 2-D. Indici dello SparseTensor concatenato.
  • Output valori_output: 1-D. Valori non vuoti dello SparseTensor concatenato.
  • Output forma_output: 1-D. Forma dello SparseTensor concatenato.

Costruttori e distruttori

SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim)

Attributi pubblici

operation
output_indices
output_shape
output_values

Attributi pubblici

operazione

Operation operation

indici_output

::tensorflow::Output output_indices

forma_output

::tensorflow::Output output_shape

valori_di_output

::tensorflow::Output output_values

Funzioni pubbliche

SparseConcat

 SparseConcat(
  const ::tensorflow::Scope & scope,
  ::tensorflow::InputList indices,
  ::tensorflow::InputList values,
  ::tensorflow::InputList shapes,
  int64 concat_dim
)