텐서의 세그먼트를 따라 합계를 계산합니다.
세그먼트에 대한 설명은 [세그먼트 섹션](https://tensorflow.org/api_docs/python/tf/math#Segmentation)을 읽어보세요.
\\(output_i = \sum_j data_j\\) 되는 텐서를 계산합니다. 여기서 합은 `j` 이상이고 `segment_ids[j] == i`가 됩니다.
주어진 세그먼트 ID `i`에 대한 합계가 비어 있으면 `output[i] = 0`입니다.
이 작업은 현재 jit_compile=True에서만 지원됩니다.
공개 방법
출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
static <T, U 확장 번호, V 확장 번호> SegmentSumV2 <T> | |
출력 <T> | 출력 () 크기가 `num_segments`인 단일 차원으로 대체되는 첫 번째 `segment_ids.rank` 차원을 제외하고 데이터와 모양이 동일합니다. |
상속된 메서드
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 SegmentSumV2 <T> 생성 ( 범위 범위, 피연산자 <T> 데이터, 피연산자 <U> 세그먼트 ID, 피연산자 <V> numSegments)
새 SegmentSumV2 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
세그먼트 ID | 크기가 '데이터'의 첫 번째 차원 크기와 동일한 1차원 텐서입니다. 값은 정렬되어야 하며 반복될 수 있습니다. 값은 'num_segments'보다 작아야 합니다. 주의: 값은 항상 CPU에서 정렬되도록 검증되며 GPU에서는 검증되지 않습니다. |
보고
- SegmentSumV2의 새 인스턴스