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 do SparseTensor concatenado.
  • Valores_de_saída Output : 1-D. Valores não vazios do SparseTensor concatenado.
  • Output output_shape: 1-D. Forma do SparseTensor 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
)