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

tf.get_variableのパラメータをテンソルに設定すると、変数はテンソルの値に初期化されます。したがって、次のコードを使用することができます：

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