aliran tensor:: operasi:: Softmax

#include <nn_ops.h>

Menghitung aktivasi softmax.

Ringkasan

Untuk setiap angkatan i dan kelas j yang kami miliki

$$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