tensorflow::ops::SparseConcat
#include <sparse_ops.h> Nối một danh sách SparseTensor theo chiều được chỉ định.
Bản tóm tắt
Phép nối được thực hiện đối với các phiên bản dày đặc của các tensor thưa 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 thứ tự số chiều tăng dần.
Tất cả các hình dạng đầu vào phải khớp nhau, ngoại trừ chiều nối chuỗi. Danh sách indices , values và shapes phải có cùng độ dài.
Hình dạng đầu ra giống hệt hình dạng đầu vào, ngoại trừ dọc theo chiều nối, nơi nó là tổng kích thước của các đầu vào dọc theo chiều đó.
Các phần tử đầu ra sẽ được sắp xếp lại để duy trì thứ tự sắp xếp theo thứ tự tăng dần của số chiều.
Thao tác này chạy trong thời gian O(M log M) , trong đó M là tổng số giá trị không rỗng trên tất cả các đầu vào. Điều này là do cần phải sắp xếp nội bộ để nối chuỗi hiệu quả trên một chiều bất kỳ.
Ví dụ, nếu concat_dim = 1 và các đầ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"
sau đó kết quả đầ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 việc thực hiện
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Luận điểm:
- phạm vi: Một đối tượng Phạm vi
- Chỉ số: 2 chiều. Chỉ số của mỗi
SparseTensorđầu vào. - Giá trị: 1 chiều. Các giá trị không rỗng của mỗi
SparseTensor. - Hình dạng: 1 chiều. Hình dạng của mỗi
SparseTensor. - concat_dim: Chiều để nối. Phải nằm trong khoảng [-rank, rank), trong đó rank là số chiều trong mỗi
SparseTensorđầu vào.
Trả về:
- Đầu
Outputoutput_indices: 2 chiều. Chỉ số củaSparseTensorđược nối. -
Outputoutput_values: 1 chiều. Các giá trị không rỗng củaSparseTensorđược nối lại. - Đầu
Outputoutput_shape: 1 chiều. Hình dạng củaSparseTensorđược nối lại.
Hàm khởi tạo và hàm 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
chỉ số đầu ra
::tensorflow::Output output_indices
hình dạng đầu ra
::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 )