6c06a88957549b1405f43361e4aa911abc56ef66,se3cnn/kernel.py,SE3Kernel,combination,#SE3Kernel#Any#,296

Before Change

                    w = weight[weight_index: weight_index + mi * mj * b_el].view(mi * mj, b_el)  // [I*J, beta]
                    weight_index += mi * mj * b_el

                    basis_kernels_ij = kij.contiguous().view(b_el, -1)  // [beta, i*j*x*y*z]

                    ker = torch.mm(w, basis_kernels_ij)  // [I*J, i*j*x*y*z]
                    ker = ker.view(mi, mj, *b_size)  // [I, J, i, j, x, y, z]

After Change

                begin_j += mj * self.dims_in[j]
            begin_i += mi * self.dims_out[i]

        assert weight_index == self.nweights
        return kernel

    def forward(self):  // pylint: disable=W
        return self.combination(self.weight)

Instances