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, ...)