Attending to Channels Using Keras and TensorFlow