Weighted VaE