75 行 pytorch+einops 中的极快胶囊网络capsnet

[www.ziwang.com]

https://github.com/arogozhnikov/readable_capsnet

75 行 pytorch+einops 中的极快胶囊网络

胶囊网络<80条线路。

来自pytorch中胶囊之间动态路由的胶囊网络的研究友好型实现。

什么是胶囊？
胶囊是一组神经元，每个神经元描述一个具有位置特征的实体。

胶囊旨在为场景形成一个“软解析树”，该场景具有由胶囊激活规范编码的置信度。