首页 > 解决方案 > 限制屏蔽区域中的卷积计算

问题描述

有谁知道 tensorflow 是否支持仅在图像的受限区域(如蒙版区域)应用 2D 卷积?(首先将卷积应用于所有图像,然后将不必要的部分屏蔽掉是行不通的,因为其目的是通过避免不必要的操作来提高性能)如果是这样,它会破坏 SIMT 还是可以在不影响 GPU 并行化的情况下运行?我必须自己在 CUDA 中实现这样的层吗?

标签: tensorflowconv-neural-network

解决方案


推荐阅读