Imported from https://github.com/chenzhenhua986/CAFFE-CapsNet