如何使用 tf.get_variable 指定对角矩阵
How to Specify a diagonal matrix using tf.get_variable
我正在尝试使用 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 来做。
提前感谢您的帮助！
如果您将 tf.get_variable 的 initializer 参数设置为张量，变量将被初始化为张量的值。因此，您可以使用以下代码：
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.Variable(tf.matrix_diag(dia_size), name=name)
b = tf.reshape(b, [-1, img_size, img_size, num_filters])
但我不能用 tf.get_variable 来做。
提前感谢您的帮助！
如果您将 tf.get_variable 的 initializer 参数设置为张量，变量将被初始化为张量的值。因此，您可以使用以下代码：
dia_size = tf.zeros((num_filters, img_size))
b = tf.matrix_diag(dia_size)
var = tf.get_variable(..., initializer=b, ...)