没有为操作MaxPoolWithArgmax定义十位数的渐变

Question

我想在TensorFlow中使用max_pool_with_argmax操作，但我得到以下错误：

LookupError: No gradient defined for operation 'MaxPoolWithArgmax_1' (op type: MaxPoolWithArgmax)

下面是我正在使用的一段代码max_pool_with_argmax：

BN_relu13 =   tf.nn.relu(tf.nn.batch_normalization(h_conv13,batch_mean13,batch_var13,tf.Variable(tf.zeros([64])),tf.Variable(tf.ones([64])),epsilon)) 
#max pooling 
h_pool1, argmax_1 = max_pool_2x2(BN_relu13)

这是max_pool_2x2模块：

^{pr2}$

在我建立了我的网络之后，我使用AdamOptimizer来训练我的网络。在

顺便说一下，当我使用tf.nn.max_pool（而不是tf.nn.max_pool_with_argmax）时，一切都很好。在

我在GPU上运行代码，在ubuntu14上使用python2.7。在

谢谢，阿里

Answer 1

梯度操作实际上实现了here，但不知何故没有注册here。值得就它提出一个github问题。在

同时，您可以在tutorial（仅渐变注册部分）之后自己注册op。在