import torch
from torch.nn import functional as F
import torch.nn as nn


class VAE(nn.Module):
    def __init__(self):
        # TODO: Variational Auto Encoder
        super(VAE, self).__init__()

    def forward(self, x):
        pass