繁体   English   中英

在 tensorflow 1.15 中构建 minpool 层

[英]Building a minpool layer in tensorflow 1.15

我被要求在分配新建分配FY代码在tensorflow 1分钟池层(1.15,具体而言)。 我在网上看了很多资源,但老实说我有些失落。 我的意思是,我在这里看到了平均池层的实现:

张量流中的自定义中值池

但我真的不明白那里发生了什么。 我阅读了tf.extract_image_patches的文档,并不能真正理解那里发生了什么,其余的答案也不清楚。

我知道如何用 numpy 实现这一层,所以如果有一种简单的方法来获取 tf 张量,将它传递给一个 numpy 函数并作为张量返回 - 并将这个函数作为一个层集成到我的 CNN 中 - 这将很棒。

为什么不是这样的(伪代码):

neg_input = -1*input
max_of_neg = max_pool(neg_input)
min_pooled = -1*max_pool

这里的想法是 -x 的最大值是 x 的最小值(的负数)。 所以我们取负数,对其进行最大池化,然后再次取负数。
不需要自定义池操作——除非您对 C++/CUDA 编码感到满意,否则实际上创建一个与内置池一样高性能的自定义池可能非常困难。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM