tensorflow 2.0 kerasd代码-收集整理归纳总结一 non-local

苏打水的杯子

于 2021-01-19 09:54:05 发布

阅读量453

点赞数

分类专栏：人工智能

本文链接：https://blog.csdn.net/qq_38784454/article/details/112800028

版权

人工智能专栏收录该内容

82 篇文章 46 订阅 ¥59.90 ¥99.00

订阅专栏

keras自带的Reshape层不需要写batch的维度，但是tensorflow的reshape需要完整的维度。

keras中reshape需要知道具体维度，如non-locla 复现中，在reshape成原输入大小时具体的tensor维度需要明确。不能直接将dim1和dim2用作赋值即reshape(dim1,dim2,channels)(y)不可行。

ip_shape = K.int_shape(x)
bs, dim1, dim2, channels = ip_shape
# x = tf.keras.layers.Conv2D(channels, 1, strides=1, kernel_initializer=initializer, padding='same', use_bias=False)
x1 = Reshape((-1, channels))(x)  # xi
x2 = Reshape((-1, channels))(x)  # xj
f = dot([x1, x2], axes=2)
f = Activation('softmax')(f)
g = Reshape((-1, channels))(x)
y = dot([f, g], axes=[2, 1])
y = Reshape((pow(2, i), pow(2, i), channels))(y)
x = add([y, x])