Extrahiere bestimmte Dimensionen von einem Tensor

Question

Ich versuche, die tf.image_summary Funktion in Tensorflow zu verwenden. Ich versuche, den Filter einer Faltungsschicht zu visualisieren. Der Filter ist definiert als tf.Variable(tf.constant(0.1, shape=[5, 5, 16, 32])).

Aber hier, da ich nur die letzten Filter sehen möchte, möchte ich einen Weg finden, um einen Filter der Größe [5, 5, 32] zu erhalten, indem Sie einfach den ersten Index der Dimension, die 16 war Ich benutze [:,:, 0,:], dann nehme ich an, dass ich einen Filter [5, 5, 1, 32] anstelle von [5, 5, 32] bekommen würde.

Was soll ich tun?

Answer 1

Also tf.image_summary nimmt einen Batch als Eingabe aber es erwartet 1,3 oder 4 in Bezug auf Farbkanäle.

so müssten Sie in passieren, so etwas tf.image_summary:

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