tensorflow::ops::SparseSoftmax

#include <sparse_ops.h>

Áp dụng hàm softmax cho một ND SparseTensor được xử lý theo lô.

Bản tóm tắt

Các đầu vào biểu thị một ND SparseTensor có hình dạng logic [..., B, C] (trong đó N >= 2 ), và các chỉ số được sắp xếp theo thứ tự từ điển chính tắc.

Thao tác này tương đương với việc áp dụng hàm tf.nn.softmax() thông thường cho từng ma trận con logic trong cùng có hình dạng [B, C] , nhưng với điểm khác biệt là các phần tử bằng 0 ngầm định sẽ không tham gia . Cụ thể, thuật toán tương đương với như sau:

(1) Áp dụng tf.nn.softmax() cho một chế độ xem được làm dày của mỗi ma trận con trong cùng có hình dạng [B, C] , dọc theo chiều kích thước C; (2) Che đi các vị trí ban đầu có giá trị bằng không ngầm định; (3) Chuẩn hóa lại các phần tử còn lại.

Do đó, kết quả của SparseTensor có các chỉ số khác 0 và hình dạng hoàn toàn giống nhau.

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • sp_indices: Ma trận 2 chiều NNZ x R chứa các chỉ số của các giá trị không rỗng trong SparseTensor, theo thứ tự chính tắc.
  • sp_values: 1-D. Các giá trị không rỗng NNZ tương ứng với sp_indices .
  • sp_shape: 1 chiều. Hình dạng của SparseTensor đầu vào.

Trả về:

  • Output : 1 chiều. Các giá trị NNZ cho kết quả SparseTensor .

Hàm khởi 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

SparseSoftmax 

 SparseSoftmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input sp_indices,
  ::tensorflow::Input sp_values,
  ::tensorflow::Input sp_shape
)

nút 

::tensorflow::Node * node() const

operator::tensorflow::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const