ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: 정렬되지 않은 SegmentMax

#include <math_ops.h>

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

요약

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

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

\(output_i = {j...} data[j...]\) 여기서 max는 튜플 j... 해당하는 segment_ids[j...] == i 입니다.

주어진 세그먼트 ID i 의 최대 값이 비어있는 경우 특정 숫자 유형에 대해 가능한 가장 작은 값을 output[i] = numeric_limits ::lowest() 합니다. 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 접두사 인 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 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const