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

يبحث عن قيم ومؤشرات أكبر عناصر 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 .

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

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