Python tf.linalg.band_part用法及代碼示例

複製一個張量，將每個最內層矩陣中中心帶之外的所有內容設置為零。

用法

tf.linalg.band_part(
    input, num_lower, num_upper, name=None
)

參數

input 一個Tensor。秩 k 張量。
num_lower 一個Tensor。必須是以下類型之一：int32 , int64。 0-D 張量。要保留的子對角線的數量。如果為負，則保留整個下三角形。
num_upper 一個Tensor。必須與 num_lower 具有相同的類型。 0-D 張量。要保留的超對角行數。如果為負數，則保留整個上三角。
name 操作的名稱(可選)。

一個Tensor。具有與 input 相同的類型。

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 || (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.linalg.band_part(input, 1, -1) ==> [[ 0,  1,  2, 3]
                                       [-1,  0,  1, 2]
                                       [ 0, -1,  0, 1]
                                       [ 0,  0, -1, 0]],

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

有用的特殊情況：

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

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.linalg.band_part。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法

參數

返回

例如：

有用的特殊情況：