Ok thank you very much. Along the same lines, I have a decoder that looks like this:
    self.decoder = nn.Sequential(
        nn.Linear(z_dim, 256),               # B, 256
        # I want to separate here
        View((-1, 256, 1, 1)),               # B, 256,  1,  1
        nn.ReLU(True),
        nn.ConvTranspose2d(256, 64, 4),      # B,  64,  4,  4
        nn.ReLU(True),
        nn.ConvTranspose2d(64, 64, 4, 2, 1), # B,  64,  8,  8
        nn.ReLU(True),
        nn.ConvTranspose2d(64, 32, 4, 2, 1), # B,  32, 16, 16
        nn.ReLU(True),
        nn.ConvTranspose2d(32, 32, 4, 2, 1), # B,  32, 32, 32
        nn.ReLU(True),
        nn.ConvTranspose2d(32, nc, 4, 2, 1),  # B, nc, 64, 64
    )

And I want to somehow remove that tensor I just concatenated after my 1st linear layer in the decoder. Is this likewise as simple?