اپراتور XLA Gather مستند شده در را می پیچد
https://www.tensorflow.org/xla/operation_semantics#gather
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
خروجی <T> | asOutput () دسته نمادین تانسور را برمی گرداند. |
static <T گسترش TType ، U گسترش TNumber > جمع آوری <T> | |
خروجی <T> | خروجی () |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
خروجی عمومی <T> asOutput ()
دسته نمادین تانسور را برمی گرداند.
ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.
عمومی ایستا جمع آوری <T> ایجاد ( محدوده دامنه ، عملوند <T> عملوند، عملوند <U> startIndices، عملوند <U> sliceSizes، بعد رشته Numbers، شاخص های بولی مرتب شده اند)
روش Factory برای ایجاد کلاسی که یک عملیات Gather جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
عملوند | آرایه ای که از آن جمع می کنیم. |
شاخص های شروع | آرایه ای حاوی شاخص های آغازین برش هایی که جمع آوری می کنیم. |
sliceSizes | slice_sizes[i] مرزهای برش در بعد i است. |
ابعاد اعداد | یک پروتوی سریالی xla::GatherDimensionNumbers. |
شاخص ها مرتب شده اند | Boolean نشان می دهد که آیا شاخص ها مرتب شده اند یا خیر. |
برمی گرداند
- نمونه جدیدی از Gather