텐서플로우:: ops:: 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  ]
[      ]

논거:

  • 범위: Scope 객체
  • split_dim: 0-D. 분할할 차원입니다. [0, rank(shape)) 범위에 있어야 합니다.
  • 인덱스: 2차원 텐서는 희소 텐서의 인덱스를 나타냅니다.
  • 값: 1차원 텐서는 희소 텐서의 값을 나타냅니다.
  • shape: 1차원 텐서는 희소 텐서의 형태를 나타냅니다. output indices: 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

출력 인덱스 

::tensorflow::OutputList output_indices

출력 형태 

::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
)