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

Before Change

                g = self.gate_act(g)
                begin_g = 0  // index of first scalar gate capsule

            z = y.new_empty((y.size(0), size_out, y.size(2)))
            begin_y = 0  // index of first capsule

            for n, mul in enumerate(self.repr_out):
                if mul == 0:

After Change

        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()

Instances