dòng chảy :: hoạt động :: SparseConcat

#include <sparse_ops.h>

Nối danh sách SparseTensor dọc theo thứ nguyên được chỉ định.

Tóm lược

Sự kết nối liên quan đến các phiên bản dày đặc của các tensors thưa thớt này. Giả định rằng mỗi đầu vào là một SparseTensor có các phần tử được sắp xếp theo số thứ nguyên tăng dần.

Tất cả các hình dạng của đầu vào phải khớp nhau, ngoại trừ kích thước kết hợp. Danh sách indices , valuesshapes phải có cùng độ dài.

Hình dạng đầu ra giống hệt với đầu vào ', ngoại trừ dọc theo kích thước kết hợp, trong đó nó là tổng các kích thước của đầu vào dọc theo thứ nguyên đó.

Các phần tử đầu ra sẽ được sử dụng để duy trì thứ tự sắp xếp cùng với số thứ nguyên ngày càng tăng.

Op này chạy trong thời gian O(M log M) , trong đó M là tổng số giá trị không trống trên tất cả các đầu vào. Điều này là do nhu cầu sắp xếp nội bộ để nối hiệu quả trên một thứ nguyên tùy ý.

Ví dụ: nếu concat_dim = 1 và đầu vào là

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"

thì đầu ra sẽ là

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

Về mặt đồ họa, điều này tương đương với làm

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

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • chỉ số: 2-D. Các chỉ số của mỗi đầu vào SparseTensor .
  • giá trị: 1-D. Các giá trị không trống của mỗi SparseTensor .
  • hình dạng: 1-D. Hình dạng của mỗi SparseTensor .
  • concat_dim: Kích thước để nối cùng. Phải nằm trong phạm vi [-rank, rank), trong đó rank là số thứ nguyên trong mỗi SparseTensor đầu vào.

Lợi nhuận:

  • Output output_indices: 2-D. Các chỉ số của SparseTensor nối.
  • Output : 1-D. Các giá trị không trống của SparseTensor nối.
  • Output output_shape: 1-D. Hình dạng của SparseTensor nối.

Người xây dựng và Người phá hủy

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

Thuộc tính công khai

operation
output_indices
output_shape
output_values

Thuộc tính công khai

hoạt động

Operation operation

output_indices

::tensorflow::Output output_indices

output_shape

::tensorflow::Output output_shape

giá_trị_ đầu ra

::tensorflow::Output output_values

Chức năng công cộng

SparseConcat

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