aliran tensor:: operasi:: Softmax

#include <nn_ops.h>

Menghitung aktivasi softmax.

Ringkasan

Untuk setiap batch i dan kelas j kami memiliki

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

Argumen:

  • ruang lingkup: Objek Lingkup
  • logits: 2-D dengan bentuk [batch_size, num_classes] .

Pengembalian:

  • Output : Bentuknya sama dengan logits .

Konstruktor dan Destruktor

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

Atribut publik

operation
softmax

Fungsi publik

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

Atribut publik

operasi

Operation operation

softmax

::tensorflow::Output softmax

Fungsi publik

Softmax

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

simpul

::tensorflow::Node * node() const 

operator::tensorflow::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const