TensorFlow矩陣數(shù)學(xué)函數(shù)：tf.matrix_band_part

tf.matrix_band_part 函數(shù)

matrix_band_part(
    input,
    num_lower,
    num_upper,
    name=None
)

參考指南：數(shù)學(xué)函數(shù)>矩陣數(shù)學(xué)函數(shù)

復(fù)制一個張量,將每個最內(nèi)層矩陣中的所有中心區(qū)域外的所有內(nèi)容設(shè)置為零.

該 band 部分計算如下：假設(shè) input 有 k 維 [I, J, K, ..., M, N],則輸出是具有相同形狀的張量：

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

指示器函數(shù)：

in_band(m, n) = (num_lower < 0 || (m-n) <= num_lower)) && (num_upper < 0 || (n-m) <= 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.

參數(shù)：

• input：張量.秩為 k 的張量.
• num_lower：int64 類型的張量；0-D 張量；要保持的對角線的數(shù)量；如果為負,則保留整個下三角.
• num_upper：int64 類型的張量；0-D 張量；要保留的 superdiagonals 數(shù)；如果為負,則保持整個上三角.
• name：操作的名稱(可選).

返回值：

該函數(shù)將返回一個張量,該張量與 input 具有相同的類型；與 input 具有相同形狀的秩為 k 的張量；提取的帶狀張量.