Matrix-Capsules-EM-PyTorch icon indicating copy to clipboard operation
Matrix-Capsules-EM-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon yl-1993

M步有个错误吧。

Open onbigion13 opened this issue 3 years ago • 0 comments

ConvCaps的m-step成员函数，第一行代码。

`r = r* a_in.

该函数的初值 r 是形状是 (b, B, C, 1), 而 a_in 的形状是 (b, C, 1). 不能广播，直接做哈达玛积会报错的。是不是需要对 a_in unsqueeze一下？

onbigion13 avatar Jul 09 '22 10:07 onbigion13