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 de 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 dimensión.
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 delSparseTensor
concatenado. -
Output
valores_salida: 1-D. Valores no vacíos delSparseTensor
concatenado. -
Output
forma_salida: 1-D. Forma delSparseTensor
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 )