tensorflow :: ops :: SparseSplit
#include <sparse_ops.h>
SparseTensor
를 한 차원을 따라 num_split
텐서로 분할합니다.
요약
경우] shape[split_dim]
정수 배가 아닌 num_split
. 슬라이스 [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 차원 텐서는 희소 텐서의 인덱스를 나타냅니다.
- 값 : 1-D 텐서는 희소 텐서의 값을 나타냅니다.
- 모양 : 1-D. 텐서는 희소 텐서의 모양을 나타냅니다. 출력 인덱스 : 1-D 텐서 목록은 출력 희소 텐서의 인덱스를 나타냅니다.
- num_split : 분할 할 방법의 수.
보고:
-
OutputList
output_indices -
OutputList
output_values : 1-D 텐서 목록은 출력 희소 텐서의 값을 나타냅니다. -
OutputList
output_shape : 1-D 텐서 목록은 출력 희소 텐서의 모양을 나타냅니다.
생성자와 소멸자 | |
---|---|
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
출력 _ 인덱스
::tensorflow::OutputList output_indices
output_shape
::tensorflow::OutputList output_shape
출력 _ 값
::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 )