CapsNet-Pytorch

CapsNet-Pytorch copied to clipboard

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

onbigion13

JiaQiao111

squeeze function

3

hi,i think you mix up mod and mod_sq in the squeeze function, it would be great that you check if i am right

qiaokang0

Error

3

Hi! Thank you for your code. I got the error when I run your code. Can you help please to sort out the problem? Traceback (most recent call last): File...

redhat12345