flujo tensor:: operaciones:: EscasoConcat

#include <sparse_ops.h>

Concatena una lista de SparseTensor a lo largo de la dimensión especificada.

Resumen

La concatenación se realiza con respecto a las versiones densas de estos tensores dispersos. Se supone que cada entrada es un SparseTensor cuyos elementos están ordenados a lo largo de un número de dimensión creciente.

Las formas de todas las entradas deben coincidir, excepto la dimensión concat. Las listas indices , values y shapes deben tener la misma longitud.

La forma de salida es idéntica a la de las entradas, excepto a lo largo de la dimensión concat, donde es la suma de los tamaños de las entradas a lo largo de esa dimensión.

Se recurrirá a los elementos de salida para preservar el orden de clasificación a lo largo del aumento del número de dimensiones.

Esta operación se ejecuta en tiempo O(M log M) , donde M es el número total de valores no vacíos en todas las entradas. Esto se debe a la necesidad de una clasificación interna para poder concatenar eficientemente a través de una dimensión arbitraria.

Por ejemplo, si concat_dim = 1 y las entradas son

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"

entonces la salida será

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

Gráficamente esto equivale a hacer

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

Argumentos:

  • alcance: un objeto de alcance
  • índices: 2-D. Índices de cada entrada SparseTensor .
  • valores: 1-D. Valores no vacíos de cada SparseTensor .
  • formas: 1-D. Formas de cada SparseTensor .
  • concat_dim: Dimensión a concatenar. Debe estar en el rango [-rank, ranking), donde rango es el número de dimensiones en cada SparseTensor de entrada.

Devoluciones:

  • Output índices_salida: 2-D. Índices del SparseTensor concatenado.
  • Output valores_salida: 1-D. Valores no vacíos del SparseTensor concatenado.
  • Output forma_salida: 1-D. Forma del SparseTensor concatenado.

Constructores y destructores

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

operación

Operation operation

índices_de_salida

::tensorflow::Output output_indices

forma_salida

::tensorflow::Output output_shape

valores_de_salida

::tensorflow::Output output_values

Funciones públicas

EscasoConcat

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