新增模块 该代码是一个神经网络模块的实现，主要用于处理输入数据并产生输出。具体而言，该代码有两个类：squash和MaxState。 squash函数定义了一个名为squash的操作，用于对输入进行归一化处理。该函数通过计算兴趣胶囊的模长，并根据模长计算一个标量因子。然后，将标量因子乘以兴趣胶囊，得到归一化后的兴趣胶囊作为输出。 MaxState类是一个神经网络模块，该模块的功能是将输入数据