In the provided MNIST example not all capsules are seeing activation probability

[robertp624]

Hello naturomics! Thank you for this great code that is helping me understand capsule networks. In the documentation i saw that all of the capsules are seeing activation, but when i ran the code 3 capsules were not activated. Ran the code multiple times and saw that around 2-3-4 are not having any activation probability. I've attached picture after 500 and 49500 steps and also your provided example activation chart. Could you please help me solve this issue.