TF2.0中如何获取Tensor具体值？查看TFRecord文件名详情

嘿，作为TensorFlow新手遇到这个问题太正常啦！我当初刚接触TFRecord的时候也踩过这个坑——直接打印Tensor只会显示形状和 dtype，看不到实际内容。其实有几种简单的方法能拿到具体的文件名，我给你拆解一下：

方法一：Eager模式下用.numpy()提取（最常用） #

TensorFlow 2.x默认开启Eager执行模式，这时候你可以直接调用Tensor的.numpy()方法获取底层数据。不过要注意，TF里的字符串Tensor存储的是字节类型，需要转成普通字符串：

# 假设x是你从TFRecord里解析出的样本
filename_tensor = x['image/filename']
# 将字节类型转成UTF-8字符串
actual_filename = filename_tensor.numpy().decode('utf-8')
print("文件名：", actual_filename)

方法二：Graph模式下用tf.print()打印 #

如果你的代码是在Graph模式下运行（比如用了@tf.function装饰的函数），.numpy()会失效，这时候用tf.print()就能直接输出Tensor的真实内容，它会在图执行时打印到控制台：

tf.print("当前样本文件名：", x['image/filename'])

方法三：迭代数据集时批量提取 #

如果你的x是从tf.data.Dataset批量迭代出来的元素，比如处理批量数据时，可以遍历批量里的每个Tensor元素：

# 取数据集的第一个批次
for batch in dataset.take(1):
    # 批量文件名是形状为(批量大小,)的Tensor
    batch_filenames = batch['image/filename'].numpy()
    for fname_bytes in batch_filenames:
        print("文件名：", fname_bytes.decode('utf-8'))

小提示 #

如果你的文件名Tensor是标量（单个值），直接decode就行；如果是批量的数组，记得遍历每个元素处理哦！

内容的提问来源于stack exchange，提问作者ming guang