Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: 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 với 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 sự bắt giữ 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 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 độ thưa, 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. Các 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 cộng

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 cộng

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