如何使用tf.get_variable指定对角矩阵

Question

我正在尝试使用tf.get_variable创建对角矩阵 但是我不知道怎么做！ 就像我可以创建一个像对角矩阵一样的变量：

dia_size = tf.zeros((num_filters, img_size))
b = tf.Variable(tf.matrix_diag(dia_size), name=name)
b = tf.reshape(b, [-1, img_size, img_size, num_filters])

但我不能用tf.get_variable做到这一点。

提前感谢您的帮助！

Answer 1

如果将initializer tf.get_variable参数设置为张量，则变量将初始化为张量值。因此，您可以使用以下代码：

dia_size = tf.zeros((num_filters, img_size))
b = tf.matrix_diag(dia_size)
var = tf.get_variable(..., initializer=b, ...)