Python 如何在tensorflow中用argmax注册tf.nn.max\u pool\u?

Python 如何在tensorflow中用argmax注册tf.nn.max\u pool\u?,python,tensorflow,Python,Tensorflow,我想在tensorflow中使用tf.nn.max_pool_和_argmax,但它的梯度似乎没有注册,而它已经定义了。我试着自己从这里注册,但是我有点难以理解他们在做什么。(这是我上一篇关于tf.nn.max\u pool\u和_argmax的问题的帖子) 那么,有人知道我如何用更简单的方式注册它吗?或者有人能告诉我在tensorflow中注册op的概念吗 谢谢 阿里你检查过这个链接了吗:谢谢你的链接。我在那里找到了带有arg max的ma池的寄存器op。然后我将这段代码复制到tensorfl

我想在tensorflow中使用tf.nn.max_pool_和_argmax,但它的梯度似乎没有注册,而它已经定义了。我试着自己从这里注册,但是我有点难以理解他们在做什么。(这是我上一篇关于tf.nn.max\u pool\u和_argmax的问题的帖子)

那么,有人知道我如何用更简单的方式注册它吗?或者有人能告诉我在tensorflow中注册op的概念吗

谢谢


阿里

你检查过这个链接了吗:谢谢你的链接。我在那里找到了带有arg max的ma池的寄存器op。然后我将这段代码复制到tensorflow目录中的nn_grad.py文件中。它似乎在工作!但我没有检查它的梯度是否正确。顺便说一句,谢谢。没问题,如果你确实检查了梯度,并找到一些差异报告返回给tensorflow Gitub,你是否检查了此链接:谢谢链接。我在那里找到了带有arg max的ma池的寄存器op。然后我将这段代码复制到tensorflow目录中的nn_grad.py文件中。它似乎在工作!但我没有检查它的梯度是否正确。顺便说一句,谢谢。没问题,如果你确实检查了梯度并发现一些差异,请返回tensorflow github