删除矩阵的对角线条目

Question

矩阵A_fixed与矩阵A相同，只是对角线条目设置为0。 我想最小化同时使用矩阵A_fixed和A的目标函数。

在GPU上哪种方法更快/内存效率更高？ 还是两种方法都不好？

方法1：

A_fixed = A - tf.linalg.diag(tf.linalg.diag_part(A))

方法2：

A_fixed = tf.linalg.set_diag(A, tf.zeros(A.shape[0], dtype=A.dtype))