Добавьте измерение в тензор тензорного потока

У меня есть стек изображений из 4 изображений в градациях серого, которые я хочу передать в нейронную сеть с тензорным потоком.

После прочтения моих 4 изображений в оттенках серого и преобразования их в тензор их форма (4 120 160) Когда я передаю его в нейронную сеть, я получаю сообщение об ошибке. После некоторого поиска в Google я обнаружил, что мне нужна входная форма (4 120 160,1), в которой 1 обозначает цветовой канал.

Я так и не нашел, чтобы изменить форму моего тензора таким образом. Кажется, он работает с функцией расширения затемнения, но мне немного непонятно, что она делает.