forked from: Image Transformer

...
@author narutohyper