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

텐서플로:: 작전:: 정렬되지 않은 SegmentMax

#include <math_ops.h>

텐서의 세그먼트를 따라 최대값을 계산합니다.

요약

세그먼트에 대한 설명은 세분화 섹션을 참조하십시오 .

이 연산자는 (here) 에서 찾은 정렬되지 않은 세그먼트 합계 연산자와 유사합니다. 세그먼트에 대한 합계를 계산하는 대신 다음과 같은 최대값을 계산합니다.

\(output_i = {j...} data[j...]\) 여기서 max는 튜플 j... 를 초과하므로 segment_ids[j...] == i 입니다.

주어진 세그먼트 ID i 에 대해 최대값이 비어 있으면 특정 숫자 유형에 대해 가능한 가장 작은 값을 출력합니다. output[i] = numeric_limits ::lowest() output[i] = numeric_limits ::lowest() .

주어진 세그먼트 ID i 가 음수이면 해당 값이 삭제되고 결과에 포함되지 않습니다.

예를 들어:

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_max(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 4,  3, 3, 4],
#       [5,  6, 7, 8]]

인수:

  • 범위: 범위 개체
  • segment_ids: 모양이 data.shape 의 접두사인 텐서입니다.

보고:

  • Output : 첫 번째 segment_ids.rank 차원을 제외하고 데이터와 모양이 동일하며 크기가 num_segments 인 단일 차원으로 대체됩니다.

생성자와 소멸자

UnsortedSegmentMax (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments)

공개 속성

operation
output

공공 기능

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

공개 속성

작업

Operation operation

산출

::tensorflow::Output output

공공 기능

정렬되지 않은 SegmentMax

 UnsortedSegmentMax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

마디

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const