tf.linalg.band_part函数的参数意思

tf.linalg.band_part(.)

  • 函数功能:保留非主对角线的元素,其余位置的元素替换为0

参数描述

  • input: 输入的张量
  • num_lower :指定保留的主对角线下方的副对角线的数量,输入数值为负数时,表示下方的对角矩阵元素全部保留;
  • num_upper:指定保留的主对角线上方的副对角线的数量,输入数值为负数时,表示上方的对角矩阵元素全部保留;

参考资料

tensorflow中去上下三角矩阵

你可能感兴趣的:(Python第三方工具,tensorflow)