search

YuxianMeng / Matrix-Capsules-pytorch

A Pytorch implementation of "Matrix Capsules with EM routing"
233 stars 48 forks source link

Isn't here a bug？ #10

Open onbigion13 opened 2 years ago

onbigion13 commented 2 years ago

https://github.com/YuxianMeng/Matrix-Capsules-pytorch/blob/bc9e26dae9390bc96a5420a94cbe89432a1ad10a/Capsules.py#L86

pose's shape is (b, self.B, H, W), how can it be reshaped into ( b, 16, self.B, H, W) in the next line????? why does no one point this out? no error reported????

onbigion13 commented 2 years ago

哦，明白了，pose对x的切片第二个维度是倒数self.B。那就说得通了。我的眼科问题。抱歉。