Работа с RGBA с помощью увеличения канала

Мне нужен такой gan, чтобы он мог работать с четырехканальным изображением. Прочитал информацию о том, что необходимо просто указать в input слое о том, что моё изображение будет четырёхканальным. Но с этим возник вопрос. Будет ли корректно работать сам gan или нужны дополнительные действия? Если есть какие-то примеры или статьи об этом, скиньте пожалуйста.