9c309a959052ec40cf92cf4baa3894f5118cf8c4,se3cnn/blocks/point_gated_block.py,PointGatedBlock,forward,#PointGatedBlock#,46

Before Change

        else:
            if len(difference_mat.size()) == 4:
                batch, N, _M, _ = difference_mat.size()
            if len(difference_mat.size()) == 3:
                N, _M, _ = difference_mat.size()

            size_out = sum(mul * (2 * n + 1) for n, mul in enumerate(self.repr_out))

            if self.gate_act is not None:
                g = y[:, size_out:]

After Change

        if self.scalar_act is None and self.gate_act is None:
            return y

        has_batch = difference_matrix.dim() == 4
        if not has_batch:
            difference_matrix = difference_matrix.unsqueeze(0)
            y = y.unsqueeze(0)

        batch, N, _M, _ = difference_matrix.size()

        size_out = sum(mul * (2 * l + 1) for l, mul in enumerate(self.repr_out))

Instances