есть ли способ манипулировать тензором непосредственно в keras? - PullRequest
Новая
       3

есть ли способ манипулировать тензором непосредственно в keras?

1 голос
/ 16 июня 2020

Я хочу создать слой под названием «стоимость-объем», что является основной концепцией бумаги G C - Net. https://arxiv.org/pdf/1703.04309.pdf

Стоимость: enter image description here

Здесь я нашел способ выполнить sh это с помощью тензорного потока и код: 

def cost_volume()
    d = max_d//2-1
    dp_list=[]
    elw = B_.tf.concat([fl_const,fr_const],-1)
    dp_list.append(dp_list)
    for di in range(d):
        pad = B_.tf.constant([[0,0],[0,0],[di+1,0],[0,0]],dtype = 'int32')
        pad_r = B_.tf.pad(fr_const[:,:,:-1-di,:],pad,'CONSTANT')
        elw = B_.tf.concat([fl_const,fr_const],-1)
        dp_list.append(elw)
    total = B_.tf.concat(dp_list,0)
    total_exp = B_.tf.expand_dims(total,0)
    return total_exp

Поскольку он был реализован в tenorflow, я добавил B_, который является keras.backend.tensorflow_backend.

Могу ли я использовать это непосредственно в среде tenorflow? Или есть какой-нибудь алгоритм

...