我希望计算特征向量之间的欧几里得距离,这些向量是从网络中获取的tf.Tensor。
我尝试了以下方法,但出现了错误:
'Tensor' object is not iterable
我想通过矩阵计算每行之间的距离,而不需要逐行迭代。
features, _ = mnist_net(images)
feature_matrix = np.zeros(shape=(FLAGS.batch_size,FLAGS.batch_size))
for i in range (FLAGS.batch_size):
for j in range (FLAGS.batch_size):
aa = tf.slice(features,[i,0],[1,50])
bb = tf.slice(features,[j,0],[1,50])
feature_matrix[i,j] = tf.sqrt(sum((aa-bb)**2))