fluxo tensor:: ops:: SparseConcat
#include <sparse_ops.h>
Concatena uma lista de SparseTensor
ao longo da dimensão especificada.
Resumo
A concatenação diz respeito às versões densas desses tensores esparsos. Supõe-se que cada entrada seja um SparseTensor
cujos elementos são ordenados ao longo do número de dimensão crescente.
Todas as formas das entradas devem corresponder, exceto a dimensão concat. As listas de indices
, values
e shapes
devem ter o mesmo comprimento.
O formato de saída é idêntico ao das entradas, exceto ao longo da dimensão concat, onde é a soma dos tamanhos das entradas ao longo dessa dimensão.
Os elementos de saída serão utilizados para preservar a ordem de classificação ao longo do número crescente de dimensões.
Esta operação é executada em tempo O(M log M)
, onde M
é o número total de valores não vazios em todas as entradas. Isso se deve à necessidade de uma classificação interna para concatenar com eficiência em uma dimensão arbitrária.
Por exemplo, se concat_dim = 1
e as entradas forem
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"
então a saída será
shape = [2, 7] [0, 2]: "a" [0, 4]: "d" [0, 5]: "e" [1, 0]: "b" [1, 1]: "c"
Graficamente isso é equivalente a fazer
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Argumentos:
- escopo: um objeto Escopo
- índices: 2-D. Índices de cada entrada
SparseTensor
. - valores: 1-D. Valores não vazios de cada
SparseTensor
. - formas: 1-D. Formas de cada
SparseTensor
. - concat_dim: Dimensão para concatenar. Deve estar no intervalo [-rank, rank), onde rank é o número de dimensões em cada entrada
SparseTensor
.
Retorna:
-
Output
output_indices: 2-D. Índices doSparseTensor
concatenado. - Valores_de_saída
Output
: 1-D. Valores não vazios doSparseTensor
concatenado. -
Output
output_shape: 1-D. Forma doSparseTensor
concatenado.
Construtores e Destruidores | |
---|---|
SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim) |
Atributos públicos | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Atributos públicos
operação
Operation operation
índices_de_saída
::tensorflow::Output output_indices
forma_saída
::tensorflow::Output output_shape
valores_saída
::tensorflow::Output output_values
Funções públicas
SparseConcat
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )