텐서플로:: 작전:: SparseSplit
#include <sparse_ops.h>
SparseTensor
를 한 차원을 따라 num_split
텐서로 분할합니다.
요약
shape[split_dim]
이 num_split
의 정수배가 아닌 경우. Slices [0 : shape[split_dim] % num_split]
은 하나의 추가 차원을 얻습니다. 예를 들어 split_dim = 1
이고 num_split = 2
이고 입력이
input_tensor = shape = [2, 7] [ a d e ] [b c ]경우
그래픽으로 출력 텐서는 다음과 같습니다.
output_tensor[0] = shape = [2, 4] [ a ] [b c ] output_tensor[1] = shape = [2, 3] [ d e ] [ ]
인수:
- 범위: 범위 개체
- split_dim: 0-D. 분할할 차원입니다.
[0, rank(shape))
범위에 있어야 합니다. - 인덱스: 2차원 텐서는 희소 텐서의 인덱스를 나타냅니다.
- values: 1-D 텐서는 희소 텐서의 값을 나타냅니다.
- 모양: 1-D. 텐서는 희소 텐서의 모양을 나타냅니다. 출력 인덱스: 1차원 텐서 목록은 출력 희소 텐서의 인덱스를 나타냅니다.
- num_split: 분할 방법의 수입니다.
보고:
-
OutputList
output_indices -
OutputList
output_values: 1차원 텐서 목록은 출력 희소 텐서의 값을 나타냅니다. -
OutputList
output_shape: 1차원 텐서 목록은 출력 희소 텐서의 모양을 나타냅니다.
생성자와 소멸자 | |
---|---|
SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split) |
공개 속성 | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
공개 속성
작업
Operation operation
output_index
::tensorflow::OutputList output_indices
output_shape
::tensorflow::OutputList output_shape
output_values
::tensorflow::OutputList output_values
공공 기능
SparseSplit
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )