تدفق التوتر:: العمليات:: توبك
#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) |
الهياكل | |
|---|---|
| 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
)