我得到了一个由4个灰度图像组成的图像堆栈，我想将其传递给具有tensorflow的神经网络。
在阅读了我的4张灰度图像并将其转换为张量后，它们的形状是(4120160(当我把它传递给神经网络时，我得到了一条错误消息。经过一些谷歌搜索，我发现我需要(4120160,1(的输入形状，其中1代表颜色通道。
无论如何，我还没有发现以这种方式改变张量的形状。它似乎可以使用expand-dims函数，但对我来说，理解它的作用有点神秘。
您可以使用tf.expand_dims(image, -1)。
给定张量输入，此操作在输入形状的维度索引轴处插入长度为1的维度。