Tensorflow，使用unique_with_counts函数时出错

Question

我正在tensorflow中实现K最近邻算法，并且出现以下错误：

udidDevice

跟踪显示错误发生在以下行：

adb devices

这是代码的相关部分：

tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op'UniqueWithCounts' used by node UniqueWithCounts (defined at knn.py:92) with these attrs: [T=DT_BOOL, out_idx=DT_INT32]

Answer 1

如果您查看原始错误消息，则表明您正在将tf.bool类型的张量传递给tf.unique_with_counts()操作：

  

由节点的UniqueWithCounts（在knn.py:92处定义）具有以下属性：[ T = DT_BOOL ，out_idx = DT_INT32]

此操作不支持布尔。查看您的代码，我怀疑这不是您想要的吗？