tensorflow :: ops :: SparseConcat

#include <sparse_ops.h>

Concatena un elenco di SparseTensor lungo la dimensione specificata.

Sommario

La concatenazione è rispetto alle versioni dense di questi tensori sparsi. Si presume che ogni input sia uno SparseTensor cui elementi sono ordinati in base al numero di dimensione crescente.

Le forme di tutti gli input devono corrispondere, ad eccezione della dimensione concat. Gli elenchi di indices , values e shapes devono avere la stessa lunghezza.

La forma dell'output è identica a quella degli input, tranne 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 in 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 SparseTensor input.
  • valori: 1-D. Valori non vuoti di ogni SparseTensor .
  • forme: 1-D. Forme di ogni SparseTensor .
  • concat_dim: dimensione da concatenare. Deve essere nell'intervallo [-rank, rank), dove rank è il numero di dimensioni in ogni SparseTensor input.

Ritorna:

  • Output output_indices: 2-D. Indici dello SparseTensor concatenato.
  • Output output_values: 1-D. Valori non vuoti dello SparseTensor concatenato.
  • Output output_shape: 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

output_indices

::tensorflow::Output output_indices

output_shape

::tensorflow::Output output_shape

valori_uscita

::tensorflow::Output output_values

Funzioni pubbliche

SparseConcat

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