Pytorch-CapsuleNet icon indicating copy to clipboard operation
Pytorch-CapsuleNet copied to clipboard

Published 20 hours ago • verified publisher icon jindongwang

Why need to get the mean value of the batch dimension?

Open xianxuhua opened this issue 3 years ago • 1 comments 

b_ij = b_ij + a_ij.squeeze(4).mean(dim=0, keepdim=True)

xianxuhua avatar May 11 '22 03:05 xianxuhua

i guess it is similar to effect of Batch Normalization?

NeVerDieKinG avatar Jun 15 '24 06:06 NeVerDieKinG