如何打印tensorflow中的张量数

时间:2017-05-31 02:44:59

标签: python tensorflow computer-vision

inception/inception_eval.py中,我想打印标签和logits。

我尝试了一些方法,但他们不能工作 我如何获得这些张量的数量?

  with tf.Graph().as_default():
# Get images and labels from the dataset.

# with tf.Session() as sess:
#   images, labels = image_processing.inputs(dataset)
#   print ("lable")
#   sess.run (labels.eval())
images, labels = image_processing.inputs(dataset)
# sess = tf.Session()
# print("lable")
# print (labels)
# Number of classes in the Dataset label set plus 1.
# Label 0 is reserved for an (unused) background class.
num_classes = dataset.num_classes() + 1

# Build a Graph that computes the logits predictions from the
# inference model.
logits, _ = inception.inference(images, num_classes)
print ("logits")
# print (tf.cast(logits,tf.float32).eval())
# print ("_")
# print (tf.cast(_,tf.float32).eval())

1 个答案:

答案 0 :(得分:0)

如果您要检查inception_model文件,您会看到inference function返回两个张量:

  

Logits。二维浮子张量。

     

辅助Logits。 2-D float Tensor of   方头。仅用于训练。

所以你只需要创建一个会话并在这个会话中运行这个张量。