tf.signal.overlap_and_add 抛出 Attempting to capture an EagerTensor without build a function答案

【问题标题】：tf.signal.overlap_and_add throws Attempting to capture an EagerTensor without building a functiontf.signal.overlap_and_add 抛出 Attempting to capture an EagerTensor without build a function
【发布时间】：2020-03-17 21:50:08
【问题描述】：

当我运行以下 sn-p 时：

import tensorflow as tf
from tensorflow.keras import Input, layers

segments = 1
segment_length = 64000
inputs = Input(shape=(segments, segment_length))
net = tf.reshape(inputs, (-1, 1, segment_length * segments))
net = layers.Conv1D(filters=512, kernel_size=16, strides=8, use_bias=False, padding='same', data_format='channels_first')(net)
net = tf.reshape(net, (-1,2, 512, 8000))
net = tf.transpose(net, perm=[0,1,3,2])
net = layers.Dense(16, input_shape=(512,), use_bias=False)(net)
out = tf.signal.overlap_and_add(net, 8)

我收到以下异常：

Traceback (most recent call last):
  File "model/test.py", line 12, in <module>
    out = tf.signal.overlap_and_add(net, 8)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/ops/signal/reconstruction_ops.py", line 116, in overlap_and_add
    paddings = array_ops.concat([outer_paddings, paddings], 0)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/util/dispatch.py", line 180, in wrapper
    return target(*args, **kwargs)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/ops/array_ops.py", line 1598, in concat
    return gen_array_ops.concat_v2(values=values, axis=axis, name=name)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/ops/gen_array_ops.py", line 1189, in concat_v2
    "ConcatV2", values=values, axis=axis, name=name)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 414, in _apply_op_helper
    as_ref=input_arg.is_ref)
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1409, in internal_convert_n_to_tensor
    ctx=ctx))
  File "/home/mmoran/.miniconda3/envs/convtasnet/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1307, in convert_to_tensor
    raise RuntimeError("Attempting to capture an EagerTensor without "
RuntimeError: Attempting to capture an EagerTensor without building a function.

我已经尝试将tf.signal.overlap_and_add 包装在@tf.function 中，但我得到了相同的结果。除此之外，我还尝试禁用急切模式，但是当我这样做时，tf.signal.overlap_and_add 的结果是(None, 2, None) 破坏了我的自定义损失函数，该函数使用该形状创建蒙版。我究竟做错了什么？是bug吗？

我的系统设置是：

Ubuntu 19.10
tf-nightly==2.2.0.dev20200218
tf-nightly-gpu==2.2.0.dev20200218

【问题讨论】：

标签： python python-3.x tensorflow

【解决方案1】：

我在这里并没有真正回答关于 tf.signal.overlap_and_add 的问题，但是 ConvTasnet 的重叠添加可以使用转置卷积来实现，内核大小为 16，步幅为 8。它可能是也更有效率。

【讨论】：