张量流的第二个论点是什么tf.data.过滤器（）我找不到任何文件？

Question

我最近在使用

def lie_filter(line):
    return tf.equal(line['lie_id'], 2)

在

^{pr2}$

准确的错误是TypeError: lie_filter() takes 1 positional argument but 2 were given。在

只需将函数签名改为lie_filter(line, x)就可以消除错误，过滤看起来也能正常工作。然而，它让我想知道这是什么神秘的第二个论点。在

TensorFlow manual for tf.data.filter()只指定一个参数。TensorFlow也有很多例子，其中过滤是按照我上面的尝试完成的。看看，例如，imports85.py。在

将x打印在lie_filter内生成{}。在

第二个论点是什么？我在哪里可以找到关于它的文档？在

谢谢你！在

Answer 1

好吧，当然，我在提交问题后终于拿到了。正如我所怀疑的，这是我自己做的。map()返回一个(features, label)的元组。第二个参数当然是作为张量的label。在

希望这对未来的人有所帮助：）