VLAE.py 文件源码

def decode(self, z):
        zcode = list(torch.chunk(z, self.code_dims[0], dim=1))[::-1]
        h = self.act(self.fc1(zcode[0]))
        for z, fc in zip(zcode[1:], self.decode_layers):
            h = fc(h, z)
        return self.fc2(h)