Fix input size
New test with one layer CDBN
New test with one layer
Clean
New test
Support more containers
Cleanup