naturomics / CapsNet-Tensorflow

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

Question about num_outputs #68

Open keurcien opened 6 years ago

keurcien commented 6 years ago

I'm trying to read through the definition of the class CapsLayer. Does num_outputs actually correspond to the number of capsules ? From what I understand from the following code, it looks like the number of capsules is actually stored in vec_len.

capsules = []
    for i in range(self.vec_len):
    # each capsule i: [batch_size, 6, 6, 32]
        ...
        capsules.append(caps_i)

Sorry to bother you about the documentation, it's just to have a better understanding of how capsules work. Thanks for sharing your work by the way.