slzeng_do_this_shiz.py

import torch
import pytorch_autoencoder

list_of_numbers = get them from cpp
sum = sum

model = pytorch_autoencoder.ConvCVAENet(state_dim = 2,
                                        condition_dim = (200, 200, 2),
                                        latent_dim = 3,
                                        encoder_hidden_dim = 512, 
                                        decoder_hidden_dim = 512)