machine-learning - Tensorflow中CUDNN卷积BackwardFilter的等价物是什么？

Question

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

我怎样才能将这样的代码翻译成 Tensorflow？

Answer 1

有一个名为Conv2DBackpropFilter的操作（可conv2d_backprop_filter从 Python 访问），它在代码的 C++ 部分中定义，它可能是您正在寻找的。