tf.matrix_band_part 与 torch.tril 使用
这两个函数都是用来取矩阵的某一块值的,简单使用如下
tf.matrix_band_part
tf.matrix_band_part(m, -1, 0) 表示取矩阵的下左三角,不包括斜对角。
同理,tf.matrix_band_part(m, 0, -1)表示取矩阵的右上三角,不包括斜对角。
其他操作详细见tf.matrix_band_part ,搭配transpose可灵活取值;
torch.tril
torch.tril(m, diagonal=-1) 同样表示下三角,不包括斜对角;
torch.tril(m, diagonal=0) 表示下三角,包括斜对角。
具体操作见torch.tril