A question in model.py

Thanks for your great code!
I have a small problem:
In model.py, Line 115,116, is the 'i' in dim1 needed to be ':',  or the code will change only the last channel of the feature map.