تدفق التوتر:: العمليات:: توبك

#include <nn_ops.h>

يبحث عن قيم ومؤشرات أكبر عناصر k للبعد الأخير.

ملخص

إذا كان الإدخال متجهًا (المرتبة 1)، فسيتم العثور على أكبر إدخالات k في المتجه وإخراج قيمها ومؤشراتها كمتجهات. وبالتالي فإن values[j] هي أكبر مُدخل j في input ، ومؤشرها هو indices[j] .

بالنسبة للمصفوفات (المدخلات ذات الرتبة الأعلى)، يتم حساب أعلى إدخالات k في كل صف (المتجه على طول البعد الأخير). وهكذا،

values.shape = indices.shape = input.shape[:-1] + [k]

إذا كان هناك عنصران متساويان، فسيظهر عنصر الفهرس السفلي أولاً.

الحجج:

• النطاق: كائن النطاق
• الإدخال: 1-D أو أعلى مع البعد الأخير على الأقل k .
• ك: 0-د. عدد العناصر العليا التي يجب البحث عنها على طول البعد الأخير (على طول كل صف للمصفوفات).

السمات الاختيارية (انظر Attrs ):

• تم الفرز: إذا كان صحيحًا، فسيتم فرز عناصر k الناتجة حسب القيم بترتيب تنازلي.

عائدات:

• قيم Output : أكبر العناصر k على طول كل شريحة الأبعاد الأخيرة.
• مؤشرات Output : مؤشرات values ضمن البعد الأخير input .

البنائين والمدمرين
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k)
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs)

الصفات العامة
indices	:: tensorflow::Output
operation	Operation
values	:: tensorflow::Output

وظائف ثابتة العامة
Sorted (bool x)	Attrs

الهياكل
Tensorflow:: ops:: TopK:: Attrs	محددات السمات الاختيارية لـ TopK .

الصفات العامة

المؤشرات

::tensorflow::Output indices

عملية

Operation operation

قيم

::tensorflow::Output values

الوظائف العامة

توبك

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

توبك

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

وظائف ثابتة العامة

مرتبة

Attrs Sorted(
  bool x
)