https://tongfamily.com/2019/02/04/neural-networks-and-capsule-networks/