tensorflow - 如何将 pytorch Adaptive_avg_pool2d 方法转换为 keras 或 tensorflow
问题描述
我不知道如何将 PyTorch 方法转换为 Kerasadaptive_avg_pool2d
或 TensorFlow。任何人都可以帮忙吗?PyTorch 方法是
adaptive_avg_pool2d(14,[14])
我尝试使用平均池化,重塑 Keras 中的张量,但得到了错误:
ValueError:新数组的总大小必须保持不变
解决方案
我不确定我是否理解您的问题,但在 PyTorch 中,您将空间维度传递给AdaptiveAvgPool2d
. 例如,如果您想要输出大小为 5x7,则可以使用nn.AdaptiveAvgPool2d((5,7))
.
如果你想要一个全局平均池化层,你可以使用nn.AdaptiveAvgPool2d(1)
. 在 Keras 中,您可以使用GlobalAveragePooling2D
.
对于 Keras 中的其他输出尺寸,您需要使用AveragePooling2D
,但不能直接指定输出形状。您需要根据您想要的输出形状来计算/定义pool_size
、stride
和参数。padding
如果您在计算方面需要帮助,请查看CS231n 课程的此页面。
推荐阅读
- python - CNN - 负损失值
- powershell - 为什么 Powershell ISE 控制台显示 $args 值不正确
- python - Python ImportError:无法导入名称__init__.py
- c# - Xamarin Forms - 将scrollview.Scrolled附加到函数
- r - rollapply() n 个月
- apache-flink - 在 Flink Operator 中轮询外部数据源
- python - 定时器线程和 tkInter 的问题
- typescript - 根据兄弟值的类型设置类型
- linux - OpenCV如何选择要连接的相机?
- sql-server - 将第二个表中的单行与具有多列的多个匹配项匹配