697c5b9d1b83ea199c7590e1d0ef695cfb99aba5,se3cnn/point/kernel.py,Kernel,__init__,#Kernel#,10

Before Change

                set_of_l_filters = set_of_l_filters.union(l_filters)

                // precompute the change of basis Q
                Q = [SO3.clebsch_gordan(l_out, l_in, l) for l in l_filters]
                Q = torch.cat(Q, dim=2)  // [m_out, m_in, l_filter * m_filter]
                self.register_buffer("Q_{}_{}".format(i, j), Q.type(torch.get_default_dtype()))

        // create the radial model: R+ -> R^n_path

After Change

                // create the set of all spherical harmonics orders needed
                set_of_l_filters = set_of_l_filters.union(l_filters)

                for l in l_filters:
                    // precompute the change of basis Q
                    Q = SO3.clebsch_gordan(l_out, l_in, l).type(torch.get_default_dtype())
                    self.register_buffer("Q_{}_{}_{}".format(l_out, l_in, l), Q)

        // create the radial model: R+ -> R^n_path
        // it contains the learned parameters
        self.R = RadialModel(n_path)

        self.set_of_l_filters = sorted(set_of_l_filters)

Instances