Python tf.compat.v1.layers.averagePoolig2D是否与tf.keras.layers.globalaveragePoolig2D相同?
我刚刚开始学习Tensorflow,我在Tensorflow 1.x中发现了一个例子,我认为迁移到Tensorflow 2.x非常困难,所以我决定在其原始版本中使用它 您可以在此github上找到代码: 我想将具有全局平均池层的扁平层更改为以下代码:Python tf.compat.v1.layers.averagePoolig2D是否与tf.keras.layers.globalaveragePoolig2D相同?,python,tensorflow,keras,Python,Tensorflow,Keras,我刚刚开始学习Tensorflow,我在Tensorflow 1.x中发现了一个例子,我认为迁移到Tensorflow 2.x非常困难,所以我决定在其原始版本中使用它 您可以在此github上找到代码: 我想将具有全局平均池层的扁平层更改为以下代码: def get_embeddings(support_set, h_dim, z_dim, reuse=False): net = convolution_block(support_set, h_dim) net = convo
def get_embeddings(support_set, h_dim, z_dim, reuse=False):
net = convolution_block(support_set, h_dim)
net = convolution_block(net, h_dim)
net = convolution_block(net, h_dim)
net = convolution_block(net, z_dim)
net = tf.compat.v1.layers.flatten(net)
return net
在Tensorflow 2.x上,我知道我想要使用的函数是,但在Tensorflow 1.x上找不到同名的函数。我找到了这个
AveragePoolg2D与GlobalAveragePoolg2D相同吗?如果您无法访问,请记住,您可以通过定义自己的lambda层来执行相同的操作
x = np.random.uniform(0,1, (2,224,224,3)).astype('float32')
GlobalAveragePooling2D()(x)
# the same as
Lambda(lambda x: tf.reduce_mean(x, axis=[1,2]))(x)