7f0ac6da5577ea93deab25ba7e0d6bca33e60bfb,se3_cnn/dropout.py,SE3DropoutF,compute_noise,#SE3DropoutF#Any#,30

Before Change

                noise.fill_(1)
            else:
                noise.bernoulli_(1 - self.p).div_(1 - self.p)
            noises.append(noise.repeat(1, dim, 1, 1, 1))
        self.noise = torch.cat(noises, dim=1)

    def forward(self, x):  // pylint: disable=W

After Change

                noise.fill_(1)
            else:
                noise.bernoulli_(1 - self.p).div_(1 - self.p)
            noise = noise.unsqueeze(2).expand(-1,-1,dim,-1,-1,-1).contiguous().view(N,mul*dim,1,1,1)
            noises.append(noise)
            // noises.append(noise.repeat(1, dim, 1, 1, 1)) // DIFFERENT PROBABILITIES WITHIN CAPSULE (WRONG)
        self.noise = torch.cat(noises, dim=1)

Instances