CapsNet-Tensorflow
CapsNet-Tensorflow copied to clipboard
Question about num_outputs
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.