Tensorflow tf.map_fn和Defun冲突

我正试图使用@function.Defun decorator在tensorflow中定义我自己的操作，因为我需要为它配备一个自定义渐变。但是，在我的代码中，我还需要调用tf.map_fn函数，以将特定函数应用于一批数据。为简单起见，我尝试做如下操作：

def add_func(x):
    return x+1

@function.Defun(tf.float32)
def test(a):
    return tf.map_fn(add, a)

with tf.Session() as sess:
    a = tf.ones(shape=(6,1))
    res = sess.run(test(a))

此代码在没有@function.Defun decorator的情况下运行良好，但使用decorator时，我得到以下错误：

InvalidArgumentError: 25 nodes in a cycle
     [[Node: test_8028ca0d_2 = test_8028ca0d[_device="/job:localhost/replica:0/task:0/cpu:0"](ones_4)]]

Caused by op 'test_8028ca0d_2', defined at:
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\spyder\utils\ipython\start_kernel.py", line 223, in <module>
    main()
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\spyder\utils\ipython\start_kernel.py", line 219, in main
    kernel.start()
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\kernelapp.py", line 474, in start
    ioloop.IOLoop.instance().start()
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\zmq\eventloop\ioloop.py", line 177, in start
    super(ZMQIOLoop, self).start()
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tornado\ioloop.py", line 887, in start
    handler_func(fd_obj, events)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tornado\stack_context.py", line 275, in null_wrapper
    return fn(*args, **kwargs)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\zmq\eventloop\zmqstream.py", line 440, in _handle_events
    self._handle_recv()
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\zmq\eventloop\zmqstream.py", line 472, in _handle_recv
    self._run_callback(callback, msg)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\zmq\eventloop\zmqstream.py", line 414, in _run_callback
    callback(*args, **kwargs)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tornado\stack_context.py", line 275, in null_wrapper
    return fn(*args, **kwargs)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\kernelbase.py", line 276, in dispatcher
    return self.dispatch_shell(stream, msg)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\kernelbase.py", line 228, in dispatch_shell
    handler(stream, idents, msg)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\kernelbase.py", line 390, in execute_request
    user_expressions, allow_stdin)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\ipkernel.py", line 196, in do_execute
    res = shell.run_cell(code, store_history=store_history, silent=silent)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\ipykernel\zmqshell.py", line 501, in run_cell
    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\IPython\core\interactiveshell.py", line 2717, in run_cell
    interactivity=interactivity, compiler=compiler, result=result)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\IPython\core\interactiveshell.py", line 2827, in run_ast_nodes
    if self.run_code(code, result):
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\IPython\core\interactiveshell.py", line 2881, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-19-c5e48d04d428>", line 1, in <module>
    runfile('C:/Users/Nicki/.spyder-py3/temp.py', wdir='C:/Users/Nicki/.spyder-py3')
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\spyder\utils\site\sitecustomize.py", line 866, in runfile
    execfile(filename, namespace)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\spyder\utils\site\sitecustomize.py", line 102, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)
  File "C:/Users/Nicki/.spyder-py3/temp.py", line 156, in <module>
    res = sess.run(test(a))
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tensorflow\python\framework\function.py", line 618, in __call__
    return _call(self._definition.signature, *args, **kwargs)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tensorflow\python\framework\function.py", line 271, in _call
    compute_shapes=False)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tensorflow\python\framework\ops.py", line 2240, in create_op
    original_op=self._default_original_op, op_def=op_def)
  File "C:\Users\Nicki\Anaconda3\lib\site-packages\tensorflow\python\framework\ops.py", line 1128, in __init__
    self._traceback = _extract_stack()

InvalidArgumentError (see above for traceback): 25 nodes in a cycle
     [[Node: test_8028ca0d_2 = test_8028ca0d[_device="/job:localhost/replica:0/task:0/cpu:0"](ones_4)]]

InvalidArgumentError:一个循环中有25个节点
[[Node:test_8028ca0d_2=test_8028ca0d[\u device=“/job:localhost/replica:0/task:0/cpu:0”]（一个/u 4）]]
由op“测试”8028ca0d_2”引起，定义为：
文件“C:\Users\Nicki\Anaconda3\lib\site packages\spyder\utils\ipython\start\u kernel.py”，第223行，在
main（）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\spyder\utils\ipython\start\u kernel.py”，主目录第219行
kernel.start（）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\kernelapp.py”，第474行，在开始处
ioloop.ioloop.instance（）.start（）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\zmq\eventloop\ioloop.py”，第177行，在开始处
super（ZMQIOLoop，self）.start（）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tornado\ioloop.py”，第887行，在开始处
handler_func（fd_obj，事件）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tornado\stack\u context.py”，第275行，空包装
返回fn（*args，**kwargs）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\zmq\eventloop\zmqstream.py”，第440行，位于事件句柄中
self.\u handle\u recv（）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\zmq\eventloop\zmqstream.py”，第472行，位于\u handle\u recv
self.\u运行\u回调（回调，消息）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\zmq\eventloop\zmqstream.py”，第414行，在运行回调中
回调（*args，**kwargs）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tornado\stack\u context.py”，第275行，空包装
返回fn（*args，**kwargs）
dispatcher中第276行的文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\kernelbase.py”
返回self.dispatch\u shell（流，消息）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\kernelbase.py”，第228行，在dispatch\u shell中
处理程序（流、标识、消息）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\kernelbase.py”，第390行，在执行请求中
用户\u表达式，允许\u stdin）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\ipkernel.py”，第196行，在do\u execute中
res=shell.run\u单元格（代码，store\u history=store\u history，silent=silent）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\ipykernel\zmqshell.py”，第501行，在运行单元中
返回超级（ZMQInteractiveShell，self）。运行单元格（*args，**kwargs）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\IPython\core\interactiveshell.py”，第2717行，位于运行单元中
交互性=交互性，编译器=编译器，结果=结果）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\IPython\core\interactiveshell.py”，第2827行，在运行节点中
如果自我运行代码（代码、结果）：
文件“C:\Users\Nicki\Anaconda3\lib\site packages\IPython\core\interactiveshell.py”，第2881行，运行代码
exec（代码对象、self.user\u全局、self.user\n）
文件“”，第1行，在
运行文件（'C:/Users/Nicki/.spyder-py3/temp.py'，wdir='C:/Users/Nicki/.spyder-py3'）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\spyder\utils\site\sitecustomize.py”，第866行，在runfile中
execfile（文件名、命名空间）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\spyder\utils\site\sitecustomize.py”，第102行，在execfile中
exec（编译（f.read（），文件名，'exec'），命名空间）
文件“C:/Users/Nicki/.spyder-py3/temp.py”，第156行，在
res=sess.run（测试（a））
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tensorflow\python\framework\function.py”，第618行，在调用中__
返回呼叫（self.\u definition.signature，*args，**kwargs）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tensorflow\python\framework\function.py”，第271行，在调用中
计算（形状=假）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tensorflow\python\framework\ops.py”，第2240行，位于create\u op
初始值=自身值。\默认值\初始值，初始值=初始值）
文件“C:\Users\Nicki\Anaconda3\lib\site packages\tensorflow\python\framework\ops.py”，第1128行，在\uuu init中__
self.\u traceback=\u extract\u stack（）
InvalidArgumentError（回溯见上文）：一个循环中有25个节点
[[Node:test_8028ca0d_2=test_8028ca0d[\u device=“/job:localhost/replica:0/task:0/cpu:0”]（一个/u 4）]]

为了结束这个问题的循环，我们发现问题存在于旧版本的TensorFlow（0.12或更早版本）中。升级到TensorFlow 1.0解决了这个问题。

听起来您在

Defun

实现中发现了一个bug。我已将此作为GitHub问题发布，我们将继续关注：谢谢：）我知道Defun实现仍处于试验阶段，但据我所知，这是为自定义操作配备渐变并在gpu上运行的唯一方法。这绝对是我们应该支持的（或提供更好的错误）！我的同事需要更多的信息来解决这个问题，您是否能够ping GitHub问题来结束这个循环？谢谢