感觉PrimaryCaps层，朋友你的理解和原文有点差异。

[onbigion13]

你的输出是8个1152为的胶囊，但是原文想表达的应该是1152个8维的向量，虽然整体标量个数一样，但是你的squash函数就成了对一个1152为的函数的作用了，文章肯定是希望对一个8维向量作用。个人拙见。