Method forward

src/encoding/networks.py:222–224

(self, vol, x, aabb=None)

        return h
    
    def forward(self, vol, x, aabb=None):
        feat_map = self.encode(vol)
        return self.decode(x, feat_map, aabb=aabb)


class AutoEncoderGroupPBR(nn.Module):