私のCNNのフィルタの重みを視覚化したいと思います。サイズはheight
x width
x input
x output
です。TensorBoardのconv2dフィルタを視覚化するimage_summary
しかし、TensorBoardはX channels
batches
X height
X width
形状のテンソルであることimage_summaryを必要とします。
フィルタの重みを正しい形式に変換するにはどうすればよいですか?
いくつかのコンテキスト:
W1 = tf.Variable(tf.random_normal([5, 5, 1, 64]), name='W1')
conv = tf.nn.conv2d(x, W1, strides=[1, 1, 1, 1], padding='SAME')