Update README.md

Update README.md

Update README.md

Update README.md

using @ to matmul

fux bugs

fix the softmax bug

change parameter col

change epochs

Update capsnet.py

simplified the code

Update README.md

Update README.md

Update README.md

Update README.md