36eef6a9be308b137a106ea84c5553e3a6691333,se3cnn/point/operations.py,PairConvolution,forward,#PairConvolution#,28

Before Change

        rb = geometry.unsqueeze(1)  // [batch, 1, b, xyz]
        ra = geometry.unsqueeze(2)  // [batch, a, 1, xyz]
        k = self.kernel(rb - ra)  // [batch, a, b, 6 * i, j]
        return torch.einsum("zabij,zacij,zadij,zbcij,zbdij,zcdij,zcdj->zabi",
                            (*k.split(k.size(3) // 6, 3), features))  // [batch, a, b, channel]


class ApplyKernel(torch.nn.Module):

After Change

        out += torch.einsum("zbcij,zcdj->zabi", (k4, features))  // [batch, a, b, channel]
        out += torch.einsum("zbdij,zcdj->zabi", (k5, features))  // [batch, a, b, channel]
        out += torch.einsum("zcdij,zcdj->zabi", (k6, features))  // [batch, a, b, channel]
        return out


class ApplyKernel(torch.nn.Module):
    def __init__(self, Kernel, Rs_in, Rs_out):

Instances