텐서플로우:: 작전:: N번째요소
#include <nn_ops.h>
마지막 차원에 대한 n
차 통계 값을 찾습니다.
요약
입력이 벡터(순위-1)인 경우 벡터에서 n번째로 작은 값을 찾아 해당 값을 스칼라 텐서로 출력합니다.
행렬(각각 더 높은 순위 입력)의 경우 각 행(각각 마지막 차원을 따른 벡터)에서 n번째로 작은 값인 항목을 계산합니다. 따라서,
values.shape = input.shape[:-1]
인수:
- 범위: 범위 개체
- 입력: 1차원 이상, 마지막 차원이
n+1
이상입니다. - n: 0-D. 마지막 차원을 따라(행렬의 각 행을 따라) 선택할 정렬된 벡터의 위치입니다. n의 유효한 범위는
[0, input.shape[:-1])
입니다.
선택적 속성( Attrs
참조):
- reverse: True로 설정하면 벡터에서 n번째로 큰 값을 찾고 그 반대도 마찬가지입니다.
보고:
-
Output
: 각 마지막 차원 조각에 대한n
차 통계입니다.
생성자와 소멸자 | |
---|---|
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n) | |
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n, const NthElement::Attrs & attrs) |
공개 속성 | |
---|---|
operation | |
values |
공공 기능 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
공개 정적 함수 | |
---|---|
Reverse (bool x) |
구조체 | |
---|---|
텐서플로우:: ops:: NthElement:: Attrs | NthElement 에 대한 선택적 속성 설정자입니다. |
공개 속성
작업
Operation operation
가치
::tensorflow::Output values
공공 기능
N번째요소
NthElement( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input n )
N번째요소
NthElement( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input n, const NthElement::Attrs & attrs )
마디
::tensorflow::Node * node() const
연산자::텐서플로우::입력
operator::tensorflow::Input() const
연산자::텐서플로우::출력
operator::tensorflow::Output() const
공개 정적 함수
뒤집다
Attrs Reverse( bool x )