Tensorflow 中的 CUDNN convolutionBackwardFilter 相当于什么？
What is the equivalent of CUDNN convolutionBackwardFilter in Tensorflow?
我正在尝试移植一些在 Tensorflow 上使用 CUDNN 的代码。其中大部分都非常容易移植，但我在将 cudnnConvolutionBackwardFilter 移植到 Tensorflow 时遇到了问题。我尝试使用 conv2d 和 conv2d_transpose，但其中 none 适用于此配置。这意味着我有 [N, V, V, C] 和 [N, F, F, K] 形式的输入，我需要 [V-F, V-F, C, K] 形式的输出。我虽然想做一个循环来完成所有的卷积并对结果求和，但我没有找到在张量流图中做这样一个循环的方法。
如何将此类代码转换为 Tensorflow？
有一个名为 Conv2DBackpropFilter 的 Op（可通过 Python 中的 conv2d_backprop_filter 访问），它是在代码的 C++ 部分中定义的，它可能就是您要查找的内容。
我正在尝试移植一些在 Tensorflow 上使用 CUDNN 的代码。其中大部分都非常容易移植，但我在将 cudnnConvolutionBackwardFilter 移植到 Tensorflow 时遇到了问题。我尝试使用 conv2d 和 conv2d_transpose，但其中 none 适用于此配置。这意味着我有 [N, V, V, C] 和 [N, F, F, K] 形式的输入，我需要 [V-F, V-F, C, K] 形式的输出。我虽然想做一个循环来完成所有的卷积并对结果求和，但我没有找到在张量流图中做这样一个循环的方法。
如何将此类代码转换为 Tensorflow？
有一个名为 Conv2DBackpropFilter 的 Op（可通过 Python 中的 conv2d_backprop_filter 访问），它是在代码的 C++ 部分中定义的，它可能就是您要查找的内容。