tf.linalg.band_part 和 tf.matrix_band_part

这俩是一个函数
函数参数有
(
input,
num_lower,
num_upper,
name=None
)
num_lower控制主对角线下侧保留几条band，band就是和主对角线平行的，离主对角线由近及远的斜着的数据，值为-1表示全部保留
num_upper控制主对角线上侧保留几条band
比如输入的矩阵数据为：
0000就是主对角线

 0,  1,  2, 3
-1,  0,  1, 2
-2, -1,  0, 1
-3, -2, -1, 0

num_lower=1 num_upper=-1

 0,  1,  2, 3
-1,  0,  1, 2
 0, -1,  0, 1
 0,  0, -1, 0

num_lower=2 num_upper=1

 0,  1,  0, 0
-1,  0,  1, 0
-2, -1,  0, 1
 0, -2, -1, 0

说白了这个函数就是控制矩阵主对角线为中心，保留两侧多少条数据

原文链接: https://www.cnblogs.com/lxzbky/p/15977970.html

欢迎关注

微信关注下方公众号，第一时间获取干货硬货；公众号内回复【pdf】免费获取数百本计算机经典书籍;

也有高质量的技术群，里面有嵌入式、搜广推等BAT大佬

tf.linalg.band_part 和 tf.matrix_band_part

原创文章受到原创版权保护。转载请注明出处：https://www.ccppcoding.com/archives/399913

非原创文章文中已经注明原地址，如有侵权，联系删除

关注公众号【高性能架构探索】，第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处！

tf.linalg.band_part 和 tf.matrix_band_part

相关推荐