Как использовать один и тот же фильтр тензорного потока на нескольких каналах - PullRequest
Новая
       78

Как использовать один и тот же фильтр тензорного потока на нескольких каналах

0 голосов
/ 29 мая 2020

Я пытаюсь использовать 2 фильтра для набора входных данных и убедиться, что каждый фильтр действует для каждого входа. См. Прикрепленное изображение визуализация тензоров .

Я пытался реализовать функцию, которая выполняет это: 

@tf.function
def filter_1(inputs,filters):


    # grab some useful statistics about the inputs
    batch_size = inputs.shape[0]
    n_samp = inputs.shape[2]
    n_taps = h.shape[2]

    outputs = tf.nn.conv2d(
                tf.reshape(inputs,(batch_size,1,n_samp,2)),
                tf.reshape(filters_casted,(1,n_taps,1,2)),
                1,
                'SAME',
                data_format='NCHW'
              )

    return outputs

Но это дает очень неверные результаты. Может кто-нибудь попробовать создать функцию, которая выполняет то, что показано на картинке визуализация тензоров ?

...