任何人都知道pytorchadaptive_avg_pool2d的算法,例如
adaptive_avg_pool2d(image,[14,14])
这样的问题: 我想在keras神经网络中做同样的事情,对于任何给定输入,都想获得14 * 14的输出。有什么建议吗?
答案 0 :(得分:1)
我认为这在Keras中不存在。您可以获取输入的维度,然后除以14得到所需的pool_size。
例如,如果您的输入为28x28,则可以使用:
keras.layers.AveragePooling2D(pool_size=(2, 2), strides=None, padding='valid', data_format=None)