टेंसरफ़्लो:: ऑप्स:: विरल विभाजन

#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  ]
[      ]

तर्क:

  • स्कोप: एक स्कोप ऑब्जेक्ट
  • स्प्लिट_डिम: 0-डी। वह आयाम जिसके साथ विभाजित करना है। [0, rank(shape)) की सीमा में होना चाहिए।
  • सूचकांक: 2-डी टेंसर विरल टेंसर के सूचकांकों का प्रतिनिधित्व करता है।
  • मान: 1-डी टेंसर विरल टेंसर के मूल्यों का प्रतिनिधित्व करता है।
  • आकार: 1-डी। टेंसर विरल टेंसर के आकार का प्रतिनिधित्व करता है। आउटपुट इंडेक्स: 1-डी टेंसर की एक सूची आउटपुट स्पैस टेंसर के इंडेक्स का प्रतिनिधित्व करती है।
  • num_split: विभाजित करने के तरीकों की संख्या।

रिटर्न:

  • OutputList सूची
  • 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

output_values

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