tensorflow::ops::SparseConcat

#include <sparse_ops.h>

Menggabungkan daftar SparseTensor sepanjang dimensi yang ditentukan.

Ringkasan

Penggabungan dilakukan terhadap versi padat dari tensor jarang ini. Diasumsikan bahwa setiap input adalah SparseTensor yang elemen-elemennya diurutkan berdasarkan nomor dimensi yang meningkat.

Semua bentuk input harus sama, kecuali dimensi penggabungan. Daftar indices , values , dan shapes harus memiliki panjang yang sama.

Bentuk output identik dengan bentuk input, kecuali pada dimensi gabungan (concat), di mana bentuknya merupakan jumlah ukuran input pada dimensi tersebut.

Elemen keluaran akan diurutkan ulang untuk mempertahankan urutan sortir seiring dengan peningkatan nomor dimensi.

Operasi ini berjalan dalam waktu O(M log M) , di mana M adalah jumlah total nilai yang tidak kosong di seluruh input. Hal ini disebabkan oleh perlunya pengurutan internal agar dapat menggabungkan secara efisien di seluruh dimensi sembarang.

Misalnya, jika concat_dim = 1 dan inputnya adalah

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"

maka hasilnya akan menjadi

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

Secara grafis, ini setara dengan melakukan

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

Argumen:

  • Cakupan: Objek Cakupan
  • indeks: 2-D. Indeks dari setiap input SparseTensor .
  • Nilai: 1-D. Nilai tidak kosong dari setiap SparseTensor .
  • Bentuk: 1-D. Bentuk dari setiap SparseTensor .
  • concat_dim: Dimensi yang akan digabungkan. Harus berada dalam rentang [-rank, rank), di mana rank adalah jumlah dimensi dalam setiap SparseTensor input.

Mengembalikan:

  • Output output_indices: 2-D. Indeks dari SparseTensor yang digabungkan.
  • Output output_values: 1-D. Nilai-nilai yang tidak kosong dari SparseTensor yang digabungkan.
  • Output output_shape: 1-D. Bentuk SparseTensor yang digabungkan.

Konstruktor dan Destruktor

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

Atribut publik

operation
output_indices
output_shape
output_values

Atribut publik

operasi

Operation operation

indeks_keluaran

::tensorflow::Output output_indices

bentuk_keluaran

::tensorflow::Output output_shape

nilai_keluaran

::tensorflow::Output output_values

Fungsi publik

SparseConcat

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