Method __init__

src/shape_assembly/models/encoder/vn_layers.py:168–170

(self, in_channels)

class VNMaxPool(nn.Module):
    def __init__(self, in_channels):
        super(VNMaxPool, self).__init__()
        self.map_to_dir = nn.Linear(in_channels, in_channels, bias=False)

    def forward(self, x):
        '''