텐서플로우:: 작전:: SparseSoftmax

#include <sparse_ops.h>

일괄 처리된 ND SparseTensor 에 소프트맥스를 적용합니다.

요약

입력은 논리적 형태 [..., B, C] (여기서 N >= 2 )와 표준 사전순으로 정렬된 인덱스를 갖는 ND SparseTensor를 나타냅니다.

이 연산은 모양이 [B, C] 인 각 가장 안쪽 논리 부분행렬에 일반 tf.nn.softmax() 적용하는 것과 동일하지만 암시적으로 0인 요소는 참여하지 않는다는 점을 알아냅니다. 특히 알고리즘은 다음과 같습니다.

(1) tf.nn.softmax() 크기 C 차원을 따라 [B, C] 모양의 가장 안쪽 부분 행렬 각각의 조밀화된 뷰에 적용합니다. (2) 원래의 암시적 0 위치를 마스크합니다. (3) 나머지 요소를 다시 정규화합니다.

따라서 SparseTensor 결과는 정확히 동일한 0이 아닌 인덱스와 모양을 갖습니다.

인수:

  • 범위: 범위 개체
  • sp_indices: 2-D. SparseTensor에서 비어 있지 않은 값의 인덱스가 있는 NNZ x R 행렬(표준 순서).
  • sp_values: 1-D. sp_indices 에 해당하는 비어 있지 않은 NNZ 값입니다.
  • sp_shape: 1-D. 입력 SparseTensor의 모양입니다.

보고:

  • Output : 1D. SparseTensor 결과에 대한 NNZ 값입니다.

생성자와 소멸자

SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape)

공개 속성

operation
output

공공 기능

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

공개 속성

작업

Operation operation

산출

::tensorflow::Output output

공공 기능

SparseSoftmax

 SparseSoftmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input sp_indices,
  ::tensorflow::Input sp_values,
  ::tensorflow::Input sp_shape
)

마디

::tensorflow::Node * node() const

연산자::텐서플로우::입력

 operator::tensorflow::Input() const

연산자::텐서플로우::출력

 operator::tensorflow::Output() const