کلاس نهایی عمومی BroadcastHelper
اپراتور کمکی برای انجام پخش به سبک XLA
با استفاده از قوانین پخش XLA برای اپراتورهای باینری، «lhs» و «rhs» را با افزودن ابعاد 1 به هر یک از «lhs» و «rhs» که رتبههای پایینتری دارند، در یک رتبه پخش میکند.
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
static <T TType > BroadcastHelper <T> را گسترش می دهد | |
خروجی <T> | lhsOutput () تانسور LHS پخش شده |
خروجی <T> | rhsOutput () تانسور RHS پخش شده |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
مقدار ثابت: "XlaBroadcastHelper"
روش های عمومی
عمومی استاتیک BroadcastHelper <T> ایجاد ( scope scope، Operand <T> lhs، Operand <T> rhs، Operand <? گسترش TNumber > broadcastDims)
روش کارخانه برای ایجاد کلاسی که یک عملیات BroadcastHelper جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
lhs | تانسور ورودی LHS |
rhs | تانسور ورودی RHS |
broadcastDims | مشخصات ابعاد پخش به سبک XLA |
برمی گرداند
- نمونه جدیدی از BroadcastHelper