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)) .
  • المؤشرات: يمثل الموتر ثنائي الأبعاد مؤشرات الموتر المتفرق.
  • القيم: يمثل الموتر أحادي البعد قيم الموتر المتفرق.
  • الشكل: أحادي البعد. يمثل الموتر شكل الموتر المتفرق. مؤشرات الإخراج: قائمة من الموترات أحادية البعد تمثل مؤشرات الموترات المتفرقة الناتجة.
  • num_split: عدد طرق التقسيم.

المرتجعات:

  • OutputList output_indices
  • OutputList output_values: قائمة من الموترات أحادية البعد تمثل قيم الموترات المتفرقة الناتجة.
  • OutputList output_shape: قائمة من الموترات أحادية البعد تمثل شكل الموترات المتفرقة الناتجة.

البناة والمدمرون

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
)