เทนเซอร์โฟลว์::ออปส์::ซอฟต์แม็กซ์

#include <nn_ops.h>

คำนวณค่าแอคติเวชันซอฟต์แม็กซ์

สรุป

สำหรับแต่ละชุดข้อมูล i และชั้นเรียน j เรามี

$$softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))$$

ข้อโต้แย้ง:

  • ขอบเขต: วัตถุ ขอบเขต
  • logits: 2 มิติ รูปทรง [batch_size, num_classes]

การส่งคืนสินค้า:

  • Output : มีรูปร่างเหมือนกับ logits

ตัวสร้างและตัวทำลาย

Softmax (const :: tensorflow::Scope & scope, :: tensorflow::Input logits)

คุณลักษณะสาธารณะ

operation
softmax

หน้าที่สาธารณะ

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

คุณลักษณะสาธารณะ

การดำเนินการ

Operation operation

ซอฟต์แม็กซ์

::tensorflow::Output softmax

หน้าที่สาธารณะ

ซอฟต์แม็กซ์

 Softmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input logits
)

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ::เทนเซอร์โฟลว์::อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ::เทนเซอร์โฟลว์::เอาต์พุต

 operator::tensorflow::Output() const