Saswat Das

Fails to build in the following environment cmake command : ```cmake cmake .. -DCMAKE_TOOLCHAIN_FILE="C:\src\vcpkg\scripts\buildsystems\vcpkg.cmake" -DArrayFire_DIR="C:\Program Files\ArrayFire\v3\cmake\ArrayFireConfig.cmake" -DFL_BACKEND=CPU -G "Visual Studio 16 2019" -A x64 -DFL_BUILD_DISTRIBUTED=OFF -DFL_BUILD_APP_ASR=OFF -DFLBUILD_APP_OBJDET=OFF -DCMAKE_WINDOWS_EXPORT_ALL_SYMBOLS=1 ``` Build...

https://github.com/jindongwang/Pytorch-CapsuleNet/blob/93996119a224e9c3d274df88735f1acb36592d97/capsnet.py#L79 To make it stable, following is suggested: ```python def squash(self, input_tensor, epsilon=1e-7): squared_norm = (input_tensor ** 2 + epsilon).sum(-1, keepdim=True) output_tensor = squared_norm * input_tensor / ((1. + squared_norm)...