五年前用CUDA写了一个简单的4D卷积 http://t.cn/A6QmcnCe 当时做optical flow需要用到，Pytorch下又没有，只好自己写了一个。不敢说自己写得多好，但还是花了很多时间去优化性能，做到可以直接拿来用了。
发布于 河北
yaolubrain
五年前用CUDA写了一个简单的4D卷积 http://t.cn/A6QmcnCe 当时做optical flow需要用到，Pytorch下又没有，只好自己写了一个。不敢说自己写得多好，但还是花了很多时间去优化性能，做到可以直接拿来用了。