Method __init__

src/shape_assembly/models/encoder/vn_layers.py:243–270

(
            self,
            in_channels,
            dim=4,
            share_nonlinearity=False,
            negative_slope=0.2,
            use_rmat=False,
    )

    """VN-Invariant layer."""

    def __init__(
            self,
            in_channels,
            dim=4,
            share_nonlinearity=False,
            negative_slope=0.2,
            use_rmat=False,
    ):
        super().__init__()

        self.dim = dim
        self.use_rmat = use_rmat
        self.vn1 = VNLinearBNLeakyReLU(
            in_channels,
            in_channels // 2,
            dim=dim,
            share_nonlinearity=share_nonlinearity,
            negative_slope=negative_slope,
        )
        self.vn2 = VNLinearBNLeakyReLU(
            in_channels // 2,
            in_channels // 4,
            dim=dim,
            share_nonlinearity=share_nonlinearity,
            negative_slope=negative_slope,
        )
        self.vn_lin = conv1x1(
            in_channels // 4, 2 if self.use_rmat else 3, dim=dim)

    def forward(self, x):
        """