У меня есть объект Tensor dataset_tf со следующей спецификацией:
Tensor("loss/dense_7_loss/Const:0", shape=(214, 9), dtype=float64)
Я хочу найти индекс объекта-заполнителя y в этом объекте:
Tensor("dense_7_target:0", shape=(?, 9), dtype=float32)
Я использовал tf.where , чтобы найти индекс вышеупомянутой строки:
index=tf.where(tf.equal(dataset_tf, y))
Когда я хочу получить доступ к строке через этот индекс
row=dataset_tf[index,:]
Я вижу следующее сообщение об ошибке:
ValueError: Shapes must be equal rank, but are 2 and 0
From merging shape 0 with other shapes. for 'loss/dense_7_loss/strided_slice/stack_1' (op: 'Pack') with input shapes: [?,2], [].