我在张量流中做了一些简单的图像分割，我遇到了这个问题 . 我有两个张量：
A形状（batch_size，28,28,10） - 十个28x28图像，每个标签一个
B有形状（batch_size，） - 十个标签
我想制作形状的张量C（batch_size，28,28,1），这样C的最后一个维度就是A中B的10个中的一个 . 换句话说，我想只保留图像 . 相应的标签 .
我尝试了几种方法，但我无法使尺寸正确 .