Method forward

src/encoding/networks.py:119–121

(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 AutoEncoderGroupSkip(nn.Module):