テンソルから特定の次元を抽出する

テンソルフローでtf.image_summary関数を使用しようとしています。私は畳み込みレイヤのフィルタを視覚化しようとしています。フィルタはtf.Variable(tf.constant(0.1, shape=[5, 5, 16, 32]))と定義されています。テンソルから特定の次元を抽出する

しかし、ここでは最終フィルタのみを見たいので、16の次元の最初のインデックスを取るだけでサイズ[5、5、32]のフィルタを取得する方法を見つけたい私は[：、：、0、：]を使用し、[5、5、32]の代わりに[5、5、1、32]フィルタを取得すると仮定します。

どうすればよいですか？

出典

2016-10-03 Baiyu Chen

あなたのフィルタ変数が 'filters'と呼ばれ、あなたがスライス'フィルターを取る場合を[:,： ] 'とすると、本当にテンソルの形を取るべきです。' [5、5、32] '。 – mrry

また、一般的には、['tensorflow.squeeze']（https://www.tensorflow.org/versions/r0.11/api_docs/python/array_ops.html#squeeze）を使用して、サイズの次元を乗り越えることができます1。 – BarzinM

したがって、tf.image_summaryはバッチを入力として取り込みますが、カラーチャネルでは1,3または4が必要です。

ので、あなたは、このような何かtf.image_summaryするには渡す必要があるだろう：、0：

for i in range(int(math.floor(filter.get_shape()[4]/3))): 
    tf.image_summary(filter[:,:,:,i:i+3])

出典

2016-10-04 03:21:11 Steven

テンソルから特定の次元を抽出する

答えて

関連する問題