Embedding Neural Network Layer.

tests/test_variational_autoencoder.py

@@ -0,0 +1,34 @@
+from manim import *
+from PIL import Image
+from manim_ml.neural_network.embedding import EmbeddingLayer, GaussianDistribution
+from manim_ml.neural_network.feed_forward import FeedForwardLayer
+from manim_ml.neural_network.image import ImageLayer
+from manim_ml.neural_network.neural_network import NeuralNetwork
+
+config.pixel_height = 720
+config.pixel_width = 1280
+config.frame_height = 6.0
+config.frame_width = 6.0
+
+class GaussianScene(Scene):
+
+    def construct(self):
+        embedding_layer = EmbeddingLayer()
+        
+        image = Image.open('images/image.jpeg')
+        numpy_image = np.asarray(image)
+        # Make nn
+        neural_network = NeuralNetwork([
+            ImageLayer(numpy_image, height=1.4),
+            FeedForwardLayer(5),
+            FeedForwardLayer(3),
+            embedding_layer,
+            FeedForwardLayer(3),
+            FeedForwardLayer(5),
+            ImageLayer(numpy_image, height=1.4),
+        ])
+
+        neural_network.scale(1.3)
+
+        self.play(Create(neural_network))
+        self.play(neural_network.make_forward_pass_animation(run_time=15))