如何获取张量的值和索引数？

Question

preds = model([img_feat, ques_feat])
sorted_a = tf.sort(preds, direction='DESCENDING')
print(sorted_a[0][1])

它将打印tf.Tensor(0.35625213, shape=(), dtype=float32)。

这里我只需要数字 0.35625213 及其索引。

Answer 1

要打印值，可以将张量转换为numpy，然后打印：

import tensorflow as tf

# defining a float tensor
a = tf.constant(2.34)

# print tensor
print(a)              # output: tf.Tensor(2.34, shape=(), dtype=float32)

# convert to numpy and then print
print(a.numpy())      # output: 2.34