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

Before Change

                    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]
                    ker = ker.transpose(1, 2).contiguous()  // [I, i, J, j, x, y, z]
                    ker = ker.view(mi * self.dims_out[i], mj * self.dims_in[j], *b_size[2:])  // [I*i, J*j, x, y, z]
                    kernel[si, sj] = ker
                else:
                    kernel[si, sj] = 0

                begin_j += mj * self.dims_in[j]

After Change

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

                    ker = torch.einsum("uvb,bijxyz->uivjxyz", (w, kij)).contiguous()  // [u, i, v, j, x, y, z]
                    kernel[si, sj] = ker.view_as(kernel[si, sj])
                else:
                    kernel[si, sj] = 0

                begin_j += mj * self.dims_in[j]

Instances