可视化 tf.data 数据集时标签变成 (32,) 张量——batch 位置放错

/* MD / 富文本中的 .toc（含博客园搬家等嵌套结构）；.toc-box 在侧栏，不受影响 */
#content_views .toc,
/* 编辑器常在目录前后插入空 p（:empty 仍占 20px），一并去掉避免顶空隙 */
#content_views.markdown_views > p:empty:has(+ .toc),
#content_views.markdown_views > .toc + p:empty,
/* 富文本旧版目录标记 */
#content_views.htmledit_views #main-toc,
#content_views.htmledit_views #hr-toc,
#content_views.htmledit_views p[id*=\”-toc\”] {
display: none !important;
}
/* 目录去掉后，紧跟的首个标题不再多出一块上边距 */
#content_views.markdown_views > .toc + h1,
#content_views.markdown_views > .toc + h2,
#content_views.markdown_views > .toc + h3,
#content_views.markdown_views > .toc + h4,
#content_views.markdown_views > .toc + p:empty + h1,
#content_views.markdown_views > .toc + p:empty + h2,
#content_views.markdown_views > .toc + p:empty + h3,
#content_views.markdown_views > .toc + p:empty + h4 {
margin-top: 0 !important;
}

用 tf.data 搭好管线后想抽查几张图，结果 print(label) 打出来的是一整个批次的张量，图标题也变成一串乱码数字。这类问题几乎都与 batch() 在管线中的位置有关。

问题现象

下面这段是典型写法：先 resize、再 cache().batch(32).prefetch()，然后取九个样本画图。

import tensorflow as tf
import tensorflow_datasets as tfds
import matplotlib.pyplot as plt

train_ds, validation_ds, test_ds = tfds