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

Question

似乎没有简单的方法将值分配给Tensor的对角线。理想情况下，我正在寻找像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)

Answer 2

应将其更新为here的tf.linalg.set_diag