naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

reshape question #66

Open zh7117 opened 6 years ago

zh7117 commented 6 years ago

in capsNet init() else branch, how can the label(a placeholder with shape (batch_size,)) be reshaped to (batch_size,10,1) ?