dòng chảy căng:: ôi:: Thưa thớtSoftmax
#include <sparse_ops.h>
Áp dụng softmax cho ND SparseTensor
theo đợt.
Bản tóm tắt
Các đầu vào đại diện cho một ND SparseTensor có hình dạng logic [..., B, C]
(trong đó N >= 2
) và với các chỉ mục được sắp xếp theo thứ tự từ điển chính tắc.
Hoạt động này tương đương với việc áp dụng tf.nn.softmax()
bình thường cho mỗi ma trận con logic trong cùng có hình dạng [B, C]
, nhưng có lưu ý rằng các phần tử ngầm không tham gia . Cụ thể, thuật toán tương đương như sau:
(1) Áp dụng tf.nn.softmax()
cho chế độ xem dày đặc của từng ma trận con trong cùng có hình dạng [B, C]
, dọc theo kích thước size-C; (2) Che dấu các vị trí ngầm định ban đầu bằng 0; (3) Chuẩn hóa lại các phần tử còn lại.
Do đó, kết quả SparseTensor
có các chỉ số và hình dạng khác 0 giống hệt nhau.
Lập luận:
- phạm vi: Một đối tượng Phạm vi
- sp_indices: 2-D. Ma trận
NNZ x R
với các chỉ số của các giá trị không trống trong SparseTensor, theo thứ tự chuẩn. - giá trị sp: 1-D. Các giá trị
NNZ
không trống tương ứng vớisp_indices
. - sp_shape: 1-D. Hình dạng của SparseTensor đầu vào.
Trả về:
-
Output
: 1-D. Giá trịNNZ
cho kết quảSparseTensor
.
Hàm tạo và hàm hủy | |
---|---|
SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape) |
Thuộc tính công khai | |
---|---|
operation | |
output |
Chức năng công cộng | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Thuộc tính công khai
hoạt động
Operation operation
đầu ra
::tensorflow::Output output
Chức năng công cộng
Thưa thớtSoftmax
SparseSoftmax( const ::tensorflow::Scope & scope, ::tensorflow::Input sp_indices, ::tensorflow::Input sp_values, ::tensorflow::Input sp_shape )
nút
::tensorflow::Node * node() const
toán tử::tenorflow::Đầu vào
operator::tensorflow::Input() const
toán tử::tenorflow::Đầu ra
operator::tensorflow::Output() const