Tf.sets.intersection的错误: ValueError: Shape必须至少是等级2，但对于'DenseToDenseSetOperation_12‘却是等级1

Question

我在下面重现并隔离了我的错误。我是Tensorflow的新手。这个错误是因为我没有正确初始化idx1和idx2变量吗？如果是这样，我该怎么做？

我得到的错误是"ValueError: Shape必须至少是等级2，但对于'DenseToDenseSetOperation_12‘(op：'DenseToDenseSetOperation')，输入的形状必须是等级1: 3，3。“

import tensorflow as tf

idx1 = [1,2,3]
idx2 = [2,4,5]

intersection = tf.sets.intersection(tf.convert_to_tensor(idx1), tf.convert_to_tensor(idx2))
sess = tf.compat.v1.Session()
with sess.as_default():
  assert tf.compat.v1.get_default_session() is sess
  print(intersection.eval())

Answer 1

该错误指出您的idx变量不满足intersection方法的要求(有关rank的详细信息，请参阅docs )。例如，您不应该使用二维数组得到此错误：

idx1 = [[1, 2, 3]]
idx2 = [[2, 4, 5]]