Tensorflow中的CUDNN convolutionBackwardFilter等效于什么？

Question

我试图移植一些在Tensorflow上使用CUDNN的代码。它大部分很容易移植，但我有一个问题，将cudnnConvolutionBackwardFilter移植到Tensorflow。我尝试使用conv2d和conv2d_transpose，但它们都不适用于此配置。这意味着我的输入形式为[N, V, V, C]和[N, F, F, K]，我需要输出[V-F, V-F, C, K]。我虽然做了一个循环来完成所有的卷积和求和结果，但我没有找到在张量流图中做这样一个循环的方法。

如何将此类代码翻译成Tensorflow？

Answer 1

有一个Op称为Conv2DBackpropFilter（可从Python的conv2d_backprop_filter访问），它是在C++代码中定义的，它可能是您正在查找的内容。