tensorflow 取上（下）三角矩阵 tf.linalg.band_part

提示：tensorflow 可从已有矩阵中取三角矩阵，也可用于生成为1的三角矩阵。

函数说明

---

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

作用：
        以对角线为中心，取它的副对角线部分，其他部分用0填充。副对角线，即矩阵中除了            主对角线以外的其它对角线。

参数：

        input:输入的张量。
        num_lower:下三角矩阵保留的副对角线数量，取值为负数时全部保留，为0时全为0。
        num_upper:上三角矩阵保留的副对角线数量，取值为负数时全部保留，为0时全为0。

示例

示例1

==========代码==========
import tensorflow as tf
tf.enable_eager_execution()
a=tf.constant( [[1,  2,  3, 4],
				[5,  6,