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

#include <nn_ops.h>

คำนวณการเปิดใช้งาน softmax

สรุป

สำหรับแต่ละแบทช์ i และคลาส j เรามี

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

อาร์กิวเมนต์:

  • ขอบเขต: วัตถุ ขอบเขต
  • บันทึก: 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 

ตัวดำเนินการ :: tensorflow :: อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ :: tensorflow :: เอาท์พุท

 operator::tensorflow::Output() const