Как использовать только два канала для полутоновых изображений в нейросетях Convolutional и в автоэнкодерах, а не в 3 каналах RGB

Question

У меня есть полутоновые изображения (ультразвук), и я хочу обучить их, используя CNN и автоэнкодеры, но у меня проблема с количеством каналов, которое по умолчанию равно 3, когда я использую 3 канала, сообщается об ошибке, как я могу использовать только 2 канала.

ValueError: Dimensions must be equal, but are 2 and 3 for 'loss_mse/sub' (op: 'Sub') with input shapes: [?,128,128,2], [?,128,128,3].