Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

텐서플로:: 작전:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

softmax 교차 엔트로피 비용과 역전파에 대한 기울기를 계산합니다.

요약

SoftmaxCrossEntropyWithLogits 와 달리 이 작업은 레이블 확률 행렬을 허용하지 않고 기능 행당 단일 레이블을 허용합니다. 이 레이블은 주어진 행에 대해 확률이 1.0인 것으로 간주됩니다.

입력은 확률이 아니라 로짓입니다.

인수:

  • 범위: 범위 개체
  • 기능: batch_size x num_classes 행렬
  • 레이블: [0, num_classes)에 값이 있는 batch_size 벡터. 지정된 미니배치 항목에 대한 레이블입니다.

보고:

  • Output 손실: 손실 예(batch_size 벡터)당.
  • Output 역전파: 역전파 기울기(batch_size x num_classes 행렬).

생성자와 소멸자

SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels)

공개 속성

backprop
loss
operation

공개 속성

백프롭

::tensorflow::Output backprop

손실

::tensorflow::Output loss

작업

Operation operation

공공 기능

SparseSoftmaxCrossEntropyWithLogits

 SparseSoftmaxCrossEntropyWithLogits(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input features,
  ::tensorflow::Input labels
)