如何更改最后的数组尺寸？

Question

我创建了一个numpy数组，其形状如下：

(98, 15, 64, 64, 3)

我想将其更改为

(98, 15, 64, 64, 1)

我该怎么做？ 最后一个维度是图像通道，因此3是rgb，1是灰度。我想将彩色编码表示为灰度，因为图像只是黑白颜色。