Method forward

src/shape_assembly/models/decoder/MLPDecoder.py:23–27

(self, x)

        )

    def forward(self, x):
        # x.shape: (bs,1024)
        batch_size = x.shape[0]
        f = self.fc_layers(x)
        return f.reshape(batch_size, self.np, 3)