tensorflow:: عملیات:: MatrixBandPart

یک تانسور را کپی کنید که همه چیز را در خارج از یک باند مرکزی در هر ماتریس داخلی قرار می‌دهد.

خلاصه

به صفر برساند.

بخش band به صورت زیر محاسبه می‌شود: فرض کنید input دارای k بعد [I, J, K, ..., M, N] است، آنگاه خروجی یک تانسور با همان شکل است که در آن

band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n] .

تابع نشانگر

in_band(m, n) = (num_lower < 0 || (mn) <= num_lower)) && (num_upper < 0 || (nm) <= num_upper) .

برای مثال:

# if 'input' is [[ 0,  1,  2, 3]
                 [-1,  0,  1, 2]
                 [-2, -1,  0, 1]
                 [-3, -2, -1, 0]],

tf.matrix_band_part(input, 1, -1) ==> [[ 0,  1,  2, 3]
                                       [-1,  0,  1, 2]
                                       [ 0, -1,  0, 1]
                                       [ 0,  0, -1, 0]],

tf.matrix_band_part(input, 2, 1) ==> [[ 0,  1,  0, 0]
                                      [-1,  0,  1, 0]
                                      [-2, -1,  0, 1]
                                      [ 0, -2, -1, 0]]

موارد خاص مفید:

 tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
 tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
 tf.matrix_band_part(input, 0, 0) ==> Diagonal.

استدلال‌ها:

• دامنه: یک شیء دامنه
• ورودی: تانسور رتبه k .
• num_lower: تانسور صفر-بعدی. تعداد زیرقطرهایی که باید نگه داشته شوند. اگر منفی باشد، کل مثلث پایینی را نگه دارید.
• num_upper: تانسور صفر-بعدی. تعداد ابرقطرهایی که باید نگه داشته شوند. اگر منفی باشد، کل مثلث بالایی را نگه دارید.

بازده:

• Output : تانسور رتبه k با شکل مشابه ورودی. تانسور نواری استخراج شده.

ویژگی‌های عمومی

::tensorflow::Output band

Operation operation

 MatrixBandPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input num_lower,
  ::tensorflow::Input num_upper
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const