テンソルフローで変数の複雑な初期化を行うにはどうすればよいですか？

Question

私は、CNNの重みとなる変数を持っています。形状が[kernel_width、kernel_height、in_channels、out_channels]であるとしましょう。

私は何をしたいいくつかのランダムな値で初期化することであるが、各カーネルの中央値で3×3カーネルのために、私は値が[[なりたい、例えば1

しますr、r、r]、[r、1、r]、[r、r、r]]であり、各rは任意の分布から取られた異なるランダム値である。

この形式でnumpy配列を簡単に作成できますが、変数を初期化するために任意の配列を使用する方法がわかりませんでした。

Answer 1

あなたが言ったように、最も簡単な方法は、numpyの配列でフィルタを初期化することです。あなただけの変数の作成時に、この配列を渡すことがあります。

filter_value = 0.1 * np.random.randn(3, 3, 64, 64) 
filter_value[1, 1, :, :] = 1. # set the center value to 1. 

filter = tf.Variable(filter_value, dtype=tf.float32)