如何使用 plt.imshow 显示分割图像？

Question

我有一个 (512,512,21) 的张量，其中 512 是图像的高度和宽度，21 是图像分割图中的类。

预期结果

使用的方法：

plt.imshow(t[:,:,0])

产生单色（紫色）图像。