f3e7b16ce89d4e737e7608996df2bd241d594a58,se3cnn/blocks/gated_block.py,GatedBlock,forward,#GatedBlock#,96

Before Change

        // dropout
        if self.dropout is not None:
            z = self.dropout(z)

        return z

After Change

        y = self.conv(*args, **kwargs)  // [batch, channel, ...]

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

        batch = y.size(0)
        size = y.size()[2:]

Instances