텐서플로우:: 작전:: 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
: 1-D.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