Matrix 如何使用tf.get_变量指定对角矩阵

我试图使用tf.get_变量创建一个对角矩阵 但我不知道怎么做！ 我可以做一个变量，它是一个对角矩阵，比如：

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_变量来实现它

---

提前谢谢你的帮助

如果将

tf.get_variable

的

初始值设定项

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

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

---

如果将

tf.get_variable

的

初始值设定项
参数设置为张量，则变量将初始化为张量值。因此，您可以使用以下代码：

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