bojone / Capsule

A Capsule Implement with Pure Keras
350 stars 107 forks source link

why reshape before capsule #13

Open maatici opened 4 years ago

maatici commented 4 years ago

Hi; I could'nt understand why the following line is needed in test code?

cnn = Reshape((-1, 128))(cnn)

and what does 128 correspond here?

thanks in advance