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ới sp_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