Isn't here a bug？

[onbigion13]

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]

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