텐서플로우:: 작전:: 소프트맥스

#include <nn_ops.h>

소프트맥스 활성화를 계산합니다.

요약

각 배치 i 와 클래스 j 에 대해

$$softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))$$
있습니다.

인수:

  • 범위: 범위 개체
  • 로지트: [batch_size, num_classes] 모양의 2차원.

보고:

  • 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