Извлечение определенных размеров из тензора

Я пытаюсь использовать tf.image_summaryфункцию tensorflow. Я пытаюсь визуализировать фильтр сверточного слоя. Фильтр определяется как tf.Variable(tf.constant(0.1, shape=[5, 5, 16, 32])).

Но здесь, поскольку я хочу видеть только окончательные фильтры, я хочу найти способ получить фильтр размера [5, 5, 32], просто взяв первый индекс измерения, который был 16. Если я использую [:,:, 0,:], то я предполагаю, что я получу [5, 5, 1, 32] фильтр вместо [5, 5, 32], который я хочу.

Что мне делать?

1 ответ

  1. So TF.image_summary принимает пакет в качестве входных данных, однако он ожидает 1,3 или 4 с точки зрения цветовых каналов.

    так что вам придется перейти в tf.image_summary что-то вроде этого:

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