tensorflow 如何打印“tf.data.Dataset.from_tensor_slices”的结果?

jq6vz3qz  于 2023-02-13  发布在  其他
关注(0)|答案(2)|浏览(181)

我是张流的新手,所以我尝试了官方文档中出现的每一个命令。
如何正确打印结果dataset?下面是我的示例:

import tensorflow as tf
import numpy as np
sess = tf.Session()
X = tf.constant([[[1, 2, 3], [3, 4, 5]], [[3, 4, 5], [5, 6, 7]]])
Y = tf.constant([[[11]], [[12]]])
dataset = tf.data.Dataset.from_tensor_slices((X, Y))

dataset
print type(dataset)
# print help(dataset)
# print dataset.output_classes
# print dataset.output_shapes
wixjitnu

wixjitnu1#

默认情况下,TensorFlow会构建一个图形,而不是立即执行操作。如果您需要文本值,请尝试tf.enable_eager_execution()

>>> import tensorflow as tf
>>> tf.enable_eager_execution()
>>> X = tf.constant([[[1,2,3],[3,4,5]],[[3,4,5],[5,6,7]]])
>>> Y = tf.constant([[[11]],[[12]]])
>>> dataset = tf.data.Dataset.from_tensor_slices((X, Y))
>>> for x, y in dataset:
...   print(x, y)
... 
tf.Tensor(
[[1 2 3]
 [3 4 5]], shape=(2, 3), dtype=int32) tf.Tensor([[11]], shape=(1, 1), dtype=int32)
tf.Tensor(
[[3 4 5]
 [5 6 7]], shape=(2, 3), dtype=int32) tf.Tensor([[12]], shape=(1, 1), dtype=int32)

请注意,在TensorFlow 2.x中,tf.enable_eager_execution()是默认行为,符号不存在;你可以把这句话删掉。
在TensorFlow 1.x中构建图形时,您需要创建一个Session并运行图形以获取文字值:

>>> import tensorflow as tf
>>> X = tf.constant([[[1,2,3],[3,4,5]],[[3,4,5],[5,6,7]]])
>>> Y = tf.constant([[[11]],[[12]]])
>>> dataset = tf.data.Dataset.from_tensor_slices((X, Y))
>>> tensor = dataset.make_one_shot_iterator().get_next()
>>> with tf.Session() as session:
...   print(session.run(tensor))
...
(array([[1, 2, 3],
       [3, 4, 5]], dtype=int32), array([[11]], dtype=int32))
c6ubokkw

c6ubokkw2#

在最新文档中

for element in dataset:  
  print(element)

相关问题