TypeError:如果启用了Tensor相等,则Tensor无法散列-尝试在Dataset.map()函数中操作张量时。如何解决这个问题?

时间:2019-12-15 06:18:22

标签: python python-3.x tensorflow tensorflow2.0

我是TensorFlow的新手,正在编写一些代码来解析YT8M数据集。我处于地图功能的最后阶段,我正在尝试将视频标签从其原始标签(超过4,000个)映射到其所属的第一个垂直位置(仅25个)。

以下是一些代码:

def test(self, labels, features):

    #Two dictionaries here to do the mappings. 
    labels = self.vertical_to_int[self.vertical_dict[labels]] 

    return labels, features

def redo(self): 

    #get_mapped() is just the first round of mapping that I did 
above. 
    data = self.get_mapped()

    return data.map(self.test)

我觉得我已经尽了一切努力来解决此问题。试验了不同版本的Tensorflow,使用了tf.map_fn()方法而不是尝试内联地进行,尝试执行.numpy()返回EagerTensor的numpy数组,然后对其进行操作。

我不确定从这里去哪里...

谢谢!

!!!请在下面的评论中查看更新!

0 个答案:

没有答案