Imported from https://github.com/www0wwwjs1/Matrix-Capsules-EM-Tensorflow