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

#include <sparse_ops.h>

Áp dụng softmax cho ND SparseTensor .

Tóm lược

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ỉ số được sắp xếp theo thứ tự từ điển học chính tắc.

Op 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 với bắt mà các phần tử không hoàn toàn 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í ban đầu hoàn toàn bằng không; (3) Chuẩn hóa lại các yếu tố còn lại.

Do đó, kết quả SparseTensor có các chỉ số và hình dạng giống hệt nhau.

Tranh 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 Cảm biến tần số thưa thớt, theo thứ tự chính tắc.
  • sp_values: 1-D. NNZ giá trị không trống NNZ tương ứng với sp_indices .
  • sp_shape: 1-D. Hình dạng của đầu vào SparseTensor.

Lợi nhuận:

  • Output : 1-D. Giá trị NNZ cho kết quả SparseTensor .

Người xây dựng và Người phá 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

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const