텐서플로우:: 작전:: SparseSplit

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차원 텐서는 희소 텐서의 인덱스를 나타냅니다.
• 값: 1차원 텐서는 희소 텐서의 값을 나타냅니다.
• 모양: 1-D. 텐서는 희소 텐서의 모양을 나타냅니다. 출력 인덱스: 1차원 텐서 목록은 출력 희소 텐서의 인덱스를 나타냅니다.
• num_split: 분할하는 방법의 수입니다.

보고:

• OutputList 출력_인덱스
• OutputList 출력_값: 1차원 텐서 목록은 출력 희소 텐서의 값을 나타냅니다.
• OutputList output_shape: 1차원 텐서 목록은 출력 희소 텐서의 모양을 나타냅니다.

공개 속성

Operation operation

::tensorflow::OutputList output_indices

::tensorflow::OutputList output_shape

::tensorflow::OutputList output_values

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)