Tensorflow中的CUDNN convolutionBackwardFilter等效於什麼？

我試圖移植一些在Tensorflow上使用CUDNN的代碼。它大部分很容易移植，但我有一個問題，將cudnnConvolutionBackwardFilter移植到Tensorflow。我嘗試使用conv2d和conv2d_transpose，但它們都不適用於此配置。這意味着我的輸入形式爲[N, V, V, C]和[N, F, F, K]，我需要輸出[V-F, V-F, C, K]。我雖然做了一個循環來完成所有的卷積和求和結果，但我沒有找到在張量流圖中做這樣一個循環的方法。Tensorflow中的CUDNN convolutionBackwardFilter等效於什麼？

如何將此類代碼翻譯成Tensorflow？

來源

2016-12-15 Baptiste Wicht

我見過一些叫做'Conv2DBackpropFilter'的操作，可能它做了類似於你想實現的操作嗎？ – sygi

@sygi你是對的。它不在文檔中，但可以通過'''conv2d_backprop_filter'''訪問。你能把它作爲答案嗎？ –

有一個Op稱爲Conv2DBackpropFilter（可從Python的conv2d_backprop_filter訪問），它是在C++代碼中定義的，它可能是您正在查找的內容。

來源

2016-12-15 13:38:07 sygi

Tensorflow中的CUDNN convolutionBackwardFilter等效於什麼？

回答

相關問題