Python tf.print from TF2未打印到输出

我使用的是TF2，我想在tf.data.Dataset管道中运行的函数内打印张量

这是我的代码：

将tensorflow导入为tf
导入系统
sys.stdout=open（'tf.log'，'w'）
def main（）：
##数据集生成器
#
numRows=100
indx=tf.重塑（[i+1表示范围内的i（numRows）]，[numRows，1]）
features=tf.random.uniform（[numRows，2]，minval=1，maxval=10，dtype=tf.int32）
myData=tf.concat（[indx，features]，1）
##tf.data.Dataset
#
dataset=tf.data.dataset.from_tensor_切片（myData）
##管道
#
dataset.map（myFunc）
##运行管道
#
对于数据集中的d：
打印（'--'）
def myFunc（t）：
打印（t，输出流=sys.stdout）
返回t
如果名称=“\uuuuu main\uuuuuuuu”：
main（）

但结果是，我只得到了这个：

--
--
--
--
--
--
--
--
--
--
--

---

如何使用

tf.print

将张量打印到控制台？

您没有将

map

返回的数据集分配给任何对象。简单地说

dataset = tf.data.Dataset.from_tensor_slices(myData)
dataset = dataset.map(myFunc)