将张量的对角线值设置为 0

Question

似乎没有简单的方法可以为张量的对角线赋值。理想情况下，我正在寻找像 numpy.fill_diagonal 这样的命令。

目前我通过执行以下操作来完成此操作：

tf.matrix_set_diag(
                matrix,
                tf.zeros_like(matrix.shape[0:-1]),
                name=None
            )

有没有更好的办法？

Answer 1

我想你的答案应该是：

tf.matrix_set_diag(matrix, tf.zeros(matrix.shape[0:-1]), name=None)