공개 최종 클래스 SparseSegmentSumGradV2
SparseSegmentSum에 대한 기울기를 계산합니다.
값이 "인덱스"의 고유 인덱스 수인 차원 0을 제외하고 grad와 동일한 모양의 텐서 "출력"을 반환합니다. 또한 "indices"의 해당 인덱스를 포함하는 벡터 "sorted_unique_indices"를 반환합니다.
공개 방법
static <T는 숫자 확장, U는 숫자 확장, V는 숫자 확장> SparseSegmentSumGradV2 <T, U> | |
출력 <T> | 출력 () |
출력 <U> |
상속된 메서드
공개 방법
public static SparseSegmentSumGradV2 <T, U> create ( 범위 범위, 피연산자 <T> grad, 피연산자 <U> 인덱스, 피연산자 <V> 세그먼트 ID, 피연산자 <Integer>densityOutputDim0)
새로운 SparseSegmentSumGradV2 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
졸업생 | SparseSegmentSum 작업에 전파되는 그래디언트 |
지수 | 해당 SparseSegmentSum 작업에 전달된 인덱스입니다. |
세그먼트 ID | 해당 SparseSegmentSum 작업에 전달된 세그먼트_ID입니다. |
조밀한출력Dim0 | SparseSegmentSum 작업에 전달된 "데이터"의 차원 0입니다. |
보고
- SparseSegmentSumGradV2의 새 인스턴스